K-Nearest Neighbors (KNN) in ML Python

Introduction

K-Nearest Neighbors helps us guess the label of new data by looking at the closest examples we already know.

When you want to classify emails as spam or not spam based on similar past emails.

When you want to predict if a fruit is an apple or orange by comparing its features to known fruits.

When you want to recommend movies by finding users with similar tastes.

When you want to quickly group new customers based on their shopping habits.

When you want a simple model that doesn't need training time.

Syntax

ML Python

from sklearn.neighbors import KNeighborsClassifier

model = KNeighborsClassifier(n_neighbors=k)
model.fit(X_train, y_train)
predictions = model.predict(X_test)

n_neighbors is the number of closest points to look at.

You must fit the model with training data before predicting.

Examples

Use 3 neighbors to decide the class.

ML Python

model = KNeighborsClassifier(n_neighbors=3)

Neighbors closer to the point have more influence.

ML Python

model = KNeighborsClassifier(n_neighbors=5, weights='distance')

Train the model and then predict labels for new data.

ML Python

model.fit(X_train, y_train)
predictions = model.predict(X_test)

Sample Model

This program loads iris flower data, splits it, trains a KNN model with 3 neighbors, predicts flower types on test data, and prints predictions and accuracy.

ML Python

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score

# Load example data
iris = load_iris()
X, y = iris.data, iris.target

# Split data into training and testing sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# Create KNN model with 3 neighbors
model = KNeighborsClassifier(n_neighbors=3)

# Train the model
model.fit(X_train, y_train)

# Predict on test data
predictions = model.predict(X_test)

# Calculate accuracy
accuracy = accuracy_score(y_test, predictions)

print(f"Predictions: {predictions}")
print(f"Accuracy: {accuracy:.2f}")

Important Notes

KNN works best with small to medium datasets because it compares new points to all training points.

Choosing the right number of neighbors (k) is important: too small can be noisy, too large can smooth out details.

Features should be scaled (like using normalization) for KNN to work well because it uses distance.

Summary

KNN predicts labels by looking at the closest known examples.

It's easy to use and needs no training time but can be slow with big data.

Choosing the number of neighbors and scaling data are key to good results.

Practice

1. What does the K in K-Nearest Neighbors (KNN) represent?

easy

A. The number of closest data points used to predict the label

B. The number of features in the dataset

C. The number of classes in the target variable

D. The number of training samples

Solution

1. Step 1: Understand the role of K in KNN

   K represents how many nearest neighbors the algorithm looks at to decide the label for a new point.

2. Step 2: Differentiate K from other terms

   K is not about features, classes, or total samples but specifically the count of neighbors considered.

3. Final Answer:

   The number of closest data points used to predict the label -> Option A

4. Quick Check:

   K = number of neighbors [OK]

Hint: K means neighbors counted to decide label [OK]

Common Mistakes:

• Confusing K with number of features
• Thinking K is number of classes
• Assuming K is total data points

2. Which of the following is the correct way to import KNeighborsClassifier from scikit-learn in Python?

easy

A. import sklearn.neighbors.KNeighborsClassifier

B. import KNeighborsClassifier from sklearn.neighbors

C. from sklearn import KNeighborsClassifier

D. from sklearn.neighbors import KNeighborsClassifier

Solution

1. Step 1: Recall correct Python import syntax

   In Python, the correct way to import a class from a module is 'from module import Class'.

2. Step 2: Match with scikit-learn KNN import

   The KNeighborsClassifier is inside sklearn.neighbors, so the correct import is 'from sklearn.neighbors import KNeighborsClassifier'.

3. Final Answer:

   from sklearn.neighbors import KNeighborsClassifier -> Option D

4. Quick Check:

   Correct import uses 'from ... import ...' [OK]

Hint: Use 'from module import Class' syntax [OK]

Common Mistakes:

• Using 'import Class from module' which is invalid
• Trying to import directly from sklearn without neighbors
• Using dot notation in import statement

3. What will be the predicted label for the point (3, 3) using KNN with k=3 and the following training data?

X = [[1, 2], [2, 3], [3, 4], [6, 7]]
y = ['A', 'A', 'B', 'B']

medium

A. 'B'

B. 'A'

C. 'C'

D. Error due to missing label

Solution

1. Step 1: Calculate distances from (3,3) to each training point

   Distances: to (1,2) ≈ 2.24, to (2,3) = 1, to (3,4) = 1, to (6,7) ≈ 5

2. Step 2: Find 3 nearest neighbors and their labels

   Nearest neighbors: (2,3) 'A', (3,4) 'B', (1,2) 'A'. Labels are ['A', 'B', 'A'].

3. Step 3: Determine majority label

   Label 'A' appears twice, 'B' once, so predicted label is 'A'.

4. Final Answer:

   'A' -> Option B

5. Quick Check:

   Majority vote among 3 neighbors = 'A' [OK]

Hint: Pick majority label among closest 3 points [OK]

Common Mistakes:

• Ignoring distance and picking random label
• Choosing label of closest single neighbor only
• Miscounting neighbors or labels

4. Identify the error in this KNN code snippet:

from sklearn.neighbors import KNeighborsClassifier
knn = KNeighborsClassifier(n_neighbors=3)
knn.fit(X_train, y_train)
preds = knn.predict(X_test)
print(preds)

Assuming X_train and y_train are numpy arrays but y_train contains string labels.

medium

A. Missing data scaling causes error

B. y_train must be numeric, strings cause error

C. No error, code runs fine with string labels

D. n_neighbors must be less than number of samples

Solution

1. Step 1: Check if KNN supports string labels

   KNN in scikit-learn supports string labels for classification without error.

2. Step 2: Verify other possible errors

   Data scaling is recommended but not mandatory; n_neighbors=3 is valid if training samples ≥3.

3. Final Answer:

   No error, code runs fine with string labels -> Option C

4. Quick Check:

   KNN accepts string labels directly [OK]

Hint: KNN accepts string labels; no numeric conversion needed [OK]

Common Mistakes:

• Assuming string labels cause errors
• Confusing scaling as mandatory
• Ignoring sample size for n_neighbors

5. You have a dataset with features on very different scales (e.g., age 0-100, income 0-100000). Why is it important to scale features before using KNN, and what could happen if you don't?

hard

A. Scaling ensures all features contribute equally; without it, large-scale features dominate distance calculations

B. Scaling is not needed; KNN automatically adjusts for feature scales

C. Scaling reduces training time but does not affect predictions

D. Scaling changes labels to numeric values for KNN

Solution

1. Step 1: Understand distance calculation in KNN

   KNN uses distance (like Euclidean) between points; features with larger scales affect distance more.

2. Step 2: Explain effect of unscaled features

   If features are not scaled, large-scale features (like income) dominate, making smaller-scale features (like age) ignored.

3. Step 3: Importance of scaling

   Scaling puts all features on similar scale so each contributes fairly to distance and prediction.

4. Final Answer:

   Scaling ensures all features contribute equally; without it, large-scale features dominate distance calculations -> Option A

5. Quick Check:

   Scale features to balance distances [OK]

Hint: Scale features so no one dominates distance [OK]

Common Mistakes:

• Thinking KNN auto-scales features
• Believing scaling changes labels
• Assuming scaling only affects speed