算法金 | 再见!!!KNN

大侠幸会,在下全网同名「算法金」 0 基础转 AI 上岸,多个算法赛 Top 「日更万日,让更多人享受智能乐趣」

KNN算法的工作原理简单直观,易于理解和实现,这使得它在各种应用场景中备受青睐。

我们将深入探讨KNN算法,从基本概念到实现细节,从算法优化到实际应用,我们都会一一展开。通过本文,你将了解到KNN算法的核心要点,以及如何将这一强大的工具应用到实际问题中。

第一部分:KNN算法的基本概念

定义

KNN算法,全称为K-Nearest Neighbors,是一种基于实例的学习算法,或者说是一种基于记忆的学习方法。它的核心思想是,通过一个样本的K个最近邻居的多数属于某个类别,来预测该样本的类别。

工作原理

KNN算法通过以下步骤进行分类或回归:

  1. 确定距离度量:首先确定一个距离度量方法,如欧氏距离或曼哈顿距离。
  2. 寻找最近邻居:计算待分类样本与数据集中每个样本的距离,并找出距离最近的K个样本。
  3. 决策:在分类任务中,通过多数投票法决定待分类样本的类别;在回归任务中,则通过计算K个最近邻居的属性的平均值来预测。

算法特点

KNN算法具有以下显著特点:

  • 简单性:算法原理简单,易于理解和实现。
  • 无需训练:不需要训练阶段,直接使用整个数据集进行预测。
  • 自适应性:随着数据集的变化,KNN算法可以自适应地调整其预测结果。

第二部分:KNN算法的工作原理

距离度量

在KNN算法中,距离度量是确定样本之间相似性的关键。以下是几种常用的距离度量方法:

寻找最近邻居

确定一个样本的K个最近邻居涉及以下步骤:

  1. 计算距离:对于数据集中的每个点,使用选定的距离度量计算与待分类样本的距离。
  2. 排序:根据计算出的距离对所有点进行排序。
  3. 选择邻居:选择距离最小的前K个点作为最近邻居。

多数投票法(分类任务)

在分类任务中,KNN算法通过以下步骤进行决策:

  1. 收集标签:收集K个最近邻居的类别标签。
  2. 统计:统计每个类别的出现次数。
  3. 投票:选择出现次数最多的类别作为待分类样本的预测类别。

平均值法(回归任务)

在回归任务中,KNN算法预测一个连续值,通常通过以下步骤:

  1. 收集属性值:收集K个最近邻居的属性值。
  2. 计算平均值:计算这些属性值的平均值。
  3. 预测:将平均值作为待分类样本的预测结果。

第三部分:KNN算法的优缺点

优点

  1. 易于理解和实现:KNN算法的原理简单直观,易于理解,且实现起来相对容易。
  2. 不需要训练阶段:由于KNN算法在预测时直接使用整个数据集,因此它不需要一个专门的训练阶段。
  3. 自适应性强:KNN算法能够随着数据集的更新而更新,能够适应数据的变化。
  4. 可用于非线性问题:KNN算法不假设数据的分布,因此可以用于非线性问题的分类和回归。

缺点

  1. 计算密集型:由于在每次预测时都需要计算新样本与所有训练样本之间的距离,KNN算法在大数据集上可能变得非常慢。
  2. 存储需求高:KNN算法需要存储全部数据集,因此对内存的需求较高。
  3. 维数灾难:随着特征维度的增加,距离度量可能会变得不那么有效,导致所谓的"维数灾难"。
  4. 对不平衡数据敏感:KNN算法对类别不平衡的数据集比较敏感,少数类可能会被多数类所淹没。
  5. 对噪声敏感:KNN算法对噪声数据点比较敏感,噪声点可能会对预测结果产生较大影响。

第四部分:KNN算法的适用场景与局限性

适用场景

  1. 小规模数据集:KNN算法在小规模数据集上表现良好,因为它不需要复杂的训练过程。
  2. 基线模型:作为基线模型,KNN算法可以快速提供一个简单的性能标准,用于与其他更复杂的模型进行比较。
  3. 实时决策:由于KNN算法不需要预先训练,它可以用于需要快速响应的实时决策场景。
  4. 低维数据:在特征维度不是非常高的情况下,KNN算法能够很好地工作,因为它依赖于距离度量。

局限性

  1. 高计算成本:对于大规模数据集,KNN算法在预测时的计算成本非常高。
  2. 内存消耗:由于需要存储整个数据集,KNN算法对内存的需求可能会很大。
  3. 数据不平衡问题:当数据集中某些类别的样本数量远多于其他类别时,KNN算法可能会倾向于预测多数类。
  4. 噪声敏感性:KNN算法对异常值和噪声点比较敏感,这可能会影响其预测的准确性。

第五部分:KNN算法的实现与案例

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split, cross_val_score
from sklearn.neighbors import KNeighborsClassifier
from sklearn.preprocessing import StandardScaler
from sklearn.datasets import load_iris

# 加载内置的Iris数据集
iris = load_iris()
X = iris.data
y = iris.target
target_names = iris.target_names

# 数据标准化
scaler = StandardScaler()
X = scaler.fit_transform(X)

# 数据可视化(选择前两个特征进行可视化)
plt.figure(figsize=(10, 6))
for i, label in enumerate(target_names):
    plt.scatter(X[y == i, 0], X[y == i, 1], label=label)
plt.xlabel('特征 1 (标准化)')
plt.ylabel('特征 2 (标准化)')
plt.title('Iris 数据分布')
plt.legend()
plt.show()

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

# 选择最佳K值
best_k = 1

# 使用最佳K值训练模型
best_knn = KNeighborsClassifier(n_neighbors=best_k)
best_knn.fit(X_train, y_train)
accuracy = best_knn.score(X_test, y_test)
print(f"测试集上的准确率: {accuracy:.2f}")

# 可视化测试结果(选择前两个特征进行可视化)
plt.figure(figsize=(10, 6))
y_pred = best_knn.predict(X_test)
for i, label in enumerate(target_names):
    plt.scatter(X_test[y_test == i, 0], X_test[y_test == i, 1], label=f'真实: {label}')
for i, label in enumerate(target_names):
    plt.scatter(X_test[y_pred == i, 0], X_test[y_pred == i, 1], marker='x', label=f'预测: {label}')
plt.xlabel('特征 1 (标准化)')
plt.ylabel('特征 2 (标准化)')
plt.title('测试集上的预测结果')
plt.legend()
plt.show()

看运行输出,还是挺强的

第六部分:如何选择K值

选择合适的K值是KNN算法性能的关键因素之一,如下图不同的 K值,K=3 和 K=7 结果就不一样了。

以下是一些选择K值的策略:

经验法则

选择K值时,可以遵循一些经验法则:

  • 对于分类问题,K通常是奇数,以避免平票的情况。
  • K值通常小于20,并且与数据集中的样本数量成反比。

交叉验证

交叉验证是一种更系统的方法来确定最佳的K值。以下是步骤:

  1. 划分数据:将数据集划分为训练集和测试集。
  2. 循环选择K值:对一系列可能的K值进行循环。
  3. 训练和评估:对于每个K值,使用训练集训练KNN模型,并在测试集上评估其性能。
  4. 选择最佳K值:选择在测试集上表现最好的K值。

以下是使用交叉验证选择K值的Python代码示例:

from sklearn.model_selection import cross_val_score

# 可能的K值列表
k_values = list(range(1, 21))
accuracies = []

for k in k_values:
    knn = KNeighborsClassifier(n_neighbors=k)
    # 使用交叉验证评估模型性能
    accuracy = cross_val_score(knn, X, y, cv=10).mean()
    accuracies.append(accuracy)

# 找到最佳K值
best_k = k_values[np.argmax(accuracies)]
print(f"最佳K值为:{best_k}")

数据集大小与K值的关系

  • 小数据集:选择较小的K值,因为数据点较少,每个点的影响较大。
  • 大数据集:可以选择较大的K值,因为数据点较多,每个点的影响较小。

运行后输出

啊这,各种类别交织在一起 看起来密集恐惧症都要犯了

  • 这段代码生成了一张散点图,不同门派用不同颜色表示,展示了内力和速度的标准化分布。
  • 通过循环选择不同的 K 值进行交叉验证,找到表现最好的 K 值.这张图展示了不同 K 值下的交叉验证准确率,可以看到在 K=4 时准确率最高。
  • 我们使用最佳 K 值(K=4)训练模型,并在测试集上进行预测和可视化,这张图展示了测试集上真实标签和预测标签的对比(略)
  • 模型的预测准确率较低,这主要是数据集的问题,数据类别见区分度太低了,感兴趣的大侠可以试着调一下,怎么可以获得更高的准确率。期待能在评论区见

第七部分:KNN算法的优化策略

为了提高KNN算法的性能和适用性,我们可以采取以下几种优化策略:

距离度量的选择

选择适当的距离度量对于KNN算法至关重要。根据数据的特性,可以选择不同的度量方法:

权重分配

在标准KNN中,所有邻居对预测结果的贡献是相等的。然而,我们可以根据距离的远近来分配权重,使得更近的邻居对预测结果有更大的影响。权重可以按照以下公式计算:

降维技术

高维数据会加剧"维数灾难",导致KNN算法性能下降。使用降维技术如主成分分析(PCA)可以减少特征维度,同时保留数据的主要信息:

from sklearn.decomposition import PCA

# 应用PCA进行降维
pca = PCA(n_components=2)  # 假设我们只保留两个主成分
X_pca = pca.fit_transform(X)

# 现在使用降维后的数据训练KNN模型
knn = KNeighborsClassifier(n_neighbors=3)
knn.fit(X_pca, y)

其他优化方法

  • 使用不同的距离权重:根据问题的特性,可以为不同的特征赋予不同的距离权重。
  • 动态选择K值:根据样本的局部密度动态调整K值,以适应数据的不均匀分布。

第八部分:KNN算法与其他算法的比较

在机器学习领域,选择正确的算法对于解决特定问题至关重要。KNN算法因其简单性和直观性而广受欢迎,但与其他算法相比,它也有其局限性。以下是KNN算法与几种常见算法的比较:

与决策树的比较

  • 决策树:是一种监督学习算法,可以用于分类和回归。它通过学习简单的决策规则从数据特征中推断出目标值。
  • 优点:易于理解和解释,可以处理分类和回归问题,对噪声数据具有一定的鲁棒性。
  • 缺点:容易过拟合,对于不平衡的数据集表现不佳。

与支持向量机(SVM)的比较

  • 支持向量机:是一种强大的分类器,也可以用于回归问题(称为SVR)。它通过找到数据点之间的最优边界来区分不同的类别。
  • 优点:在高维空间中表现良好,对于线性和非线性问题都有解决方案。
  • 缺点:参数选择(如惩罚参数C和核函数)对性能影响很大,计算复杂度较高。

与随机森林的比较

  • 随机森林:是一种集成学习方法,由多个决策树组成。它通过构建多个树并进行投票来提高模型的准确性和鲁棒性。
  • 优点:通常比单个决策树更准确,能够处理高维数据,对过拟合有一定的抵抗力。
  • 缺点:模型可解释性较差,训练时间可能较长。

不同场景下算法选择的建议

  • 数据集大小:对于小数据集,KNN和决策树可能更合适;对于大数据集,考虑使用SVM或随机森林。
  • 数据维度:对于低维数据,KNN可以表现良好;对于高维数据,SVM或随机森林可能更合适。
  • 模型解释性:如果需要模型具有较高的解释性,决策树可能是更好的选择。
  • 计算资源:如果计算资源有限,应考虑使用计算成本较低的算法,如KNN或决策树。

[ 抱个拳,总个结 ]

在本文中,我们深入探讨了KNN算法的各个方面,从基本概念到实现细节,再到优化策略和与其他算法的比较。KNN算法以其简单直观的原理、易于实现的特点以及在小规模和低维数据集上的良好性能,确立了其在机器学习领域中的重要地位。

核心要点总结:

  • 定义:KNN是一种基于实例的分类和回归算法,通过查找测试样本的K个最近邻居来进行预测。
  • 工作原理:算法利用距离度量来确定最近邻居,并通过多数投票法或平均值法来进行决策。
  • 优点:易于理解和实现,不需要训练阶段,自适应性强。
  • 缺点:计算密集型,存储需求高,对不平衡数据和噪声敏感。
  • 适用场景:小规模数据集,基线模型,实时决策,低维数据。
  • 优化策略:包括选择合适的距离度量,权重分配,以及应用降维技术。

KNN算法虽然简单,但在许多实际应用中表现出色,尤其是在数据集较小且特征维度较低的情况下。然而,它也存在一些局限性,如高计算成本和对噪声的敏感性,这要求我们在应用时需要谨慎考虑。

[ 算法金,碎碎念 ]

全网同名,日更万日,让更多人享受智能乐趣

如过觉得内容有价值,烦请大侠多多 分享、在看、点赞,助力算法金又猛又持久、很黄很 BL 的日更下去;

同时邀请大侠 关注、星标 算法金,围观日更万日,助你功力大增、笑傲江湖

相关推荐
古希腊掌管学习的神33 分钟前
[机器学习]XGBoost(3)——确定树的结构
人工智能·机器学习
海棠AI实验室3 小时前
AI的进阶之路:从机器学习到深度学习的演变(一)
人工智能·深度学习·机器学习
IT古董4 小时前
【机器学习】机器学习的基本分类-强化学习-策略梯度(Policy Gradient,PG)
人工智能·机器学习·分类
睡觉狂魔er4 小时前
自动驾驶控制与规划——Project 3: LQR车辆横向控制
人工智能·机器学习·自动驾驶
scan7244 小时前
LILAC采样算法
人工智能·算法·机器学习
菌菌的快乐生活5 小时前
理解支持向量机
算法·机器学习·支持向量机
爱喝热水的呀哈喽5 小时前
《机器学习》支持向量机
人工智能·决策树·机器学习
大山同学5 小时前
第三章线性判别函数(二)
线性代数·算法·机器学习
苏言の狗5 小时前
Pytorch中关于Tensor的操作
人工智能·pytorch·python·深度学习·机器学习
bastgia6 小时前
Tokenformer: 下一代Transformer架构
人工智能·机器学习·llm