详解机器学习经典模型(原理及应用)——KNN

原创已于 2025-11-02 16:02:45 修改 · 2.6k 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#机器学习 #人工智能 #算法 #python

于 2024-09-24 09:22:20 首次发布

机器学习模型专栏收录该内容

19 篇文章

订阅专栏

一、什么是KNN

K最近邻（K-Nearest Neighbors，KNN）是一种简单而强大的机器学习算法，它基于一个基本的假设：相似的样本通常在特征空间中彼此接近。KNN算法通过查找一个样本的最近K个邻居来预测该样本的类别或属性。

二、KNN算法流程

KNN没有显式的训练阶段，它只需要存储训练数据集。因此，在数据集规模和特征数量相同的条件下，KNN建模训练的速度要比其他模型快很多，基本数据集读完就等于完成了模型的训练。

1、距离度量

KNN算法使用距离度量来确定数据点之间的相似性。最常见的距离度量是欧几里得距离，也可以使用曼哈顿距离、切比雪夫距离或余弦相似度等。

2、选择K值

K是一个超参数，表示在进行决策时考虑的最近邻居的数量。K的选择对模型的性能有很大影响，一般我们会根据任务指标去确定最佳的K，使用网格搜索之类的方法进行调参即可。

3、决策

对于分类任务，KNN算法会根据K个最近邻居的类别，通过多数投票法来预测新样本的类别。对于回归任务，KNN算法会计算K个最近邻居的目标值的平均值，作为新样本的目标值。

三、KNN算法优化

KNN的预测过程中，对于每一个新样本，都会依次计算该样本与训练集中每一条样本的距离，从而选出最邻近K个样本。显然，当训练集数据量较大的时候，这种预测方式将会很慢。为了提高预测时的效率，可以构建索引，如KD树或球树，这些数据结构可以加速最近邻居的搜索过程（主流的第三方库已经内置实现方式了，只需要调整参数选择合适的优化算法即可）。

1、KD（K-Dimensional Tree）树

KD树是一种用于组织点在K维空间中的数据结构，它类似于二叉搜索树，但用于多维空间。KD树通过递归地将数据集分割成两个子集来构建树结构。KD树适用于低维数据（通常小于20维），在这些情况下，它可以非常有效地进行最近邻搜索。

（1）构建过程

在每个维度上选择一个轴，并选择该维度上的中位数作为分割点。
将数据点分为两部分：一部分在分割点的一侧，另一部分在另一侧。
递归地在每个子集上重复这个过程，每次选择不同的维度。

（2）搜索过程

从根节点开始，根据数据点的值沿着树向下搜索。
在每个节点，选择与查询点在当前维度上值更接近的子树。

2、球树

球树是一种用于组织数据点的层次聚类数据结构，它将数据点聚类成一系列嵌套的超球体（几何学概念，指高维空间中半径固定的所有点的集合）。球树适用于高维数据，因为它可以更好地处理维度灾难（随着维度增加，数据点之间的距离变得不那么重要）。

（1）构建过程

选择一个数据点作为中心，创建一个包含所有点的超球体。
将数据点分为两个子集，每个子集由一个子球体表示，这些子球体与父球体相交。
递归地在每个子集上重复这个过程，每次都选择新的中心和创建新的超球体。

（2）搜索过程

从根节点开始，检查查询点是否在当前球体内部。
如果是，继续在子球体中搜索；如果不是，跳过当前分支。

四、KNN代码实现

1、分类

from sklearn.neighbors import KNeighborsClassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler

# 加载数据集
iris = load_iris()
X = iris.data
y = iris.target

# 划分数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 特征缩放
scaler = StandardScaler()
X_train = scaler.fit_transform(X_train)
X_test = scaler.transform(X_test)

# 创建KNN分类器实例
knn = KNeighborsClassifier(n_neighbors=3)

# 训练模型
knn.fit(X_train, y_train)

# 在测试集上进行预测
y_pred = knn.predict(X_test)

# 计算准确率
accuracy = knn.score(X_test, y_test)
print(f'Accuracy: {accuracy:.2f}')

2、回归

# 导入必要的库
from sklearn.datasets import load_boston
from sklearn.neighbors import KNeighborsRegressor
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_squared_error

# 加载波士顿房价数据集
boston = load_boston()
X = boston.data
y = boston.target

# 划分数据集为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 创建 KNN 回归器实例
# n_neighbors 表示 K 的值，即考虑的最近邻居的数量
knn_reg = KNeighborsRegressor(n_neighbors=5)

# 训练模型
knn_reg.fit(X_train, y_train)

# 在测试集上进行预测
y_pred = knn_reg.predict(X_test)

# 计算均方误差（MSE）
mse = mean_squared_error(y_test, y_pred)
print(f'Mean Squared Error: {mse:.2f}')

# 计算决定系数 R^2
r2 = knn_reg.score(X_test, y_test)
print(f'R^2 Score: {r2:.2f}')