Python代码实现KNN算法

Python代码实现KNN算法

KNN(K-Nearest Neighbors)是一种常用的分类算法,它的基本思想是:对于一个未知样本,找与最近的K个已知样本,然后根据这K个样本的类别进行分类。Python中,可以使用scikit-learn库实现KNN分类算法。本文将详细讲解Python实现KNN分类算法的完整攻略,包括算法原理、Python实现过程和示例。

算法原理

KNN算法的基本思想是:对于一个未知样本,找到与其最近的K个已知样本,然后根据这K个样本的类别进行分类。KNN算法的实现过程如下:

  1. 计算未知样本每个已知样本之间的距。
  2. 选取距离最近的K个已知样本。
  3. 根据这K个样本类别进行分类,通常采用数表决法。

Python实现过程

在Python中,可以使用scikit-learn库实现KNN分类算法。以下是使用scikit-learn库实现KNN分类算法的示例代码:

from sklearn.neighbors import KNeighborsClassifier

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

# 训练模型
X_train = [[0, 0], [1, 1], [2, 2], [3, 3]]
y_train = [0, 0, 1, 1]
knn.fit(X_train, y_train)

# 测试新样本
X_test = [[1.5, 1.5], [2.5, 2.5]]
y_pred = knn.predict(X_test)

# 输出预测结果
print(y_pred)

上述代码中,首先导入KNeighborsClassifier。然后,创建KNN分类器,设置K值为3。接着,使用训练数据训练模型。最后,使用测试数据进行预测,并输出预测结果。

示例1:鸢尾花分类

假设有一组鸢尾花数据,包括花萼长度、花萼宽度、花瓣长度和花瓣宽度等特征需要使用KNN算法对其进行分类。可以使用以下代码实现:

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

# 加载数据
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.3, random_state=42)

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

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

# 预测新样本
y_pred = knn.predict(X_test)

# 输出预测准确率
print("Accuracy: ", accuracy_score(y_test, y_pred))

执行上述代码后,可以得到以下输出结果:

Accuracy:  0.9777777777777777

示例2:手写数字识别

假设有一组手写数字数据,需要使用KNN算法对其进行分类。可以使用以下代码实现:

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

# 加载数据
digits = load_digits()
X = digits.data
y = digits.target

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

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

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

# 预测新样本
y_pred = knn.predict(X_test)

# 输出预测准确率
print("Accuracy: ", accuracy_score(y_test, y_pred))

执行上述代码后,可以得到以下输出结果:

Accuracy:  0.9888888888888889

总结

本文详讲解了Python实现KNN分类算法的完整攻略,包括算法原理、Python实现过程和示例。KNN算法是一种常用的分类算法,它的基本思想是:对于一个未知样本,到与其最近的K个已样本,然后根据K个样本的类别进行分类。在Python中,可以使用scikit-learn库实现KNN分类算法,具体实现过程如上述所示。通过示例,我们看到KNN算法在实应用中的灵活性和实用。

本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Python代码实现KNN算法 - Python技术站

(0)
上一篇 2023年5月13日
下一篇 2023年5月13日

相关文章

  • python逆向微信指数爬取实现步骤

    下面是“Python逆向微信指数爬取实现步骤”的完整攻略。 步骤一:安装依赖库 在开始实现微信指数爬取之前,需要安装一些依赖库: requests:用于请求数据; execjs:用于执行JS代码; pandas:用于处理数据; numpy:用于科学计算。 你可以使用以下命令进行安装: pip install requests execjs pandas nu…

    python 2023年5月23日
    00
  • 浅谈Python中用datetime包进行对时间的一些操作

    浅谈Python中用datetime包进行对时间的一些操作 Python标准库中的datetime模块提供了多种处理日期和时间的工具。本文将介绍datetime模块中最常用的类和方法,以及如何使用这些类和方法进行日期和时间的格式化、计算、比较等操作。 datetime类 datetime类是日期和时间的基础类,它包含年、月、日、时、分、秒和微秒等信息。dat…

    python 2023年6月2日
    00
  • Python – 文件处理 – 无法将’int’对象隐式转换为str [重复]

    【问题标题】:Python – File handling – Can’t convert ‘int’ object to str implicitly [duplicate]Python – 文件处理 – 无法将’int’对象隐式转换为str [重复] 【发布时间】:2023-04-06 11:31:01 【问题描述】: 我正在尝试将冒险游戏的故事从文件读…

    Python开发 2023年4月6日
    00
  • Python内建属性getattribute拦截器使用详解

    Python内建属性getattribute拦截器使用详解 在Python中,__getattribute__方法是一个能够拦截对象属性访问的强大工具。在本文中,我们将介绍该方法的详细用法和示例。 什么是__getattribute__方法? __getattribute__方法是Python的一个内置方法,它在对象属性被访问时被调用。这个方法允许你拦截对于…

    python 2023年6月6日
    00
  • Python小白必备的8个最常用的内置函数(推荐)

    Python小白必备的8个最常用的内置函数(推荐) Python作为当前最为火热的编程语言之一,有着庞大且强大的函数库,其中内置函数就是最常用的一种。本文将介绍Python小白必须掌握的8个最常用的内置函数。 1. print() print() 是Python中最基本的输出函数,可以将任何类型的数据输出到控制台。 示例代码: print("Hel…

    python 2023年5月20日
    00
  • AUC计算方法与Python实现代码

    AUC计算方法与Python实现代码 AUC(Area Under Curve)是一种常用的分类模型评价指标,它可以用于评估分类模型的性能。在本文中我们将详细介绍AUC的计算方法,并提供两个示例,以说明如何使用Python实现AUC的计算。 AUC计算方法 AUC是ROC曲线的面积,ROC曲线是一种用于评估二分类模型性能的曲线。ROC曲的横轴是假正率(Fal…

    python 2023年5月14日
    00
  • Python协程原理全面分析

    Python 协程原理全面分析 在介绍Python协程原理之前,需要先了解一些概念: 并发:同时处理多个任务。 并行:同时处理多个任务并使它们同时运行。关注于任务的执行,强调在物理上同时运行多个任务。 同步:任务按照一定的顺序进行,只有先完成前面任务才能完成后面任务。 异步:不按照任务排定的先后顺序进行,而是根据情况随时安排执行任务。异步任务可以在等待IO的…

    python 2023年5月19日
    00
  • Python实现Word的读写改操作

    针对“Python实现Word的读写改操作”的完整攻略,可以分为以下几步: 安装Python-docx库 读取Word文件 修改Word文件内容 保存Word文件 下面详细介绍每一步: 1. 安装Python-docx库 Python-docx是一个用于读取、创建、修改Word文档的Python库,可以通过以下命令在命令行中安装: pip install p…

    python 2023年6月3日
    00
合作推广
合作推广
分享本页
返回顶部