KNN Algorithm In Machine Learning | KNN Algorithm Using Python

The K-Nearest Neighbors (KNN) algorithm is a simple yet powerful supervised machine learning algorithm used for classification and regression tasks. It is a non-parametric and instance-based learning algorithm, meaning it does not make any assumptions about the underlying data distribution and stores instances of the training data itself to make predictions. In this article, we will explore the KNN algorithm in detail, including how it works, its implementation using Python, and its strengths and weaknesses.

How KNN Works

The KNN algorithm works based on the assumption that similar data points are close to each other in the feature space. To make a prediction for a new data point, the algorithm calculates the distance between that point and all other points in the training dataset. It then selects the K nearest data points (neighbors) based on the calculated distances. Finally, it assigns the majority class label (for classification) or the average value (for regression) of these K neighbors to the new data point.

Implementation of KNN Algorithm in Python

Let’s implement the KNN algorithm for a classification task using Python and the popular machine learning library scikit-learn. First, we’ll import the necessary libraries and load a sample dataset for demonstration.

import numpy as np
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score
# Load the Iris dataset
iris = load_iris()
X =
y =

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

# Standardize the features
scaler = StandardScaler()
X_train = scaler.fit_transform(X_train)
X_test = scaler.transform(X_test)

# Initialize the KNN classifier
knn = KNeighborsClassifier(n_neighbors=3)

# Train the classifier, y_train)

# Make predictions on the test set
y_pred = knn.predict(X_test)

# Calculate the accuracy of the model
accuracy = accuracy_score(y_test, y_pred)
print(“Accuracy:”, accuracy)

In this example, we use the Iris dataset, which contains samples of three different species of iris flowers, and we aim to predict the species based on the flower’s sepal and petal measurements. We split the dataset into training and testing sets, standardize the features, initialize the KNN classifier with n_neighbors=3, train the classifier on the training set, and finally, make predictions on the test set and calculate the accuracy of the model.

Strengths of KNN Algorithm

  1. Simplicity: KNN is easy to understand and implement, making it ideal for beginners in machine learning.
  2. No Training Phase: KNN does not have a training phase, which means the model can adapt to new data quickly.
  3. Versatility: KNN can be used for both classification and regression tasks.
  4. Non-Parametric: KNN does not make any assumptions about the underlying data distribution, making it suitable for a wide range of problems.
  5. Interpretability: The algorithm is highly interpretable since the prediction is based on the actual data points in the dataset.

Weaknesses of KNN Algorithm

  1. Computational Complexity: KNN can be computationally expensive, especially for large datasets, as it requires calculating the distance between the new data point and all other data points in the dataset.
  2. Memory Usage: Since KNN stores all instances of the training data, it can consume a significant amount of memory, especially for large datasets.
  3. Sensitive to Noise and Outliers: KNN is sensitive to noise and outliers in the dataset, which can affect the performance of the model.
  4. Need for Feature Scaling: KNN requires feature scaling for accurate predictions, as it is based on the distance between data points.


The K-Nearest Neighbors algorithm is a simple yet powerful machine learning algorithm that can be used for classification and regression tasks. It is easy to understand and implement, making it ideal for beginners. However, it has its limitations, such as computational complexity, memory usage, and sensitivity to noise and outliers. Despite these limitations, KNN remains a popular choice for many machine learning tasks due to its simplicity and effectiveness.


Post Views: 0

Scroll to Top