机器学习—python+sklearn实现KNN&KD树算法

最新推荐文章于 2024-10-28 01:17:24 发布

BJUT赵亮

最新推荐文章于 2024-10-28 01:17:24 发布

阅读量4.3k

点赞数 2

分类专栏：机器学习 python代码文章标签：北京工业大学赵亮

本文链接：https://blog.csdn.net/qq_22235017/article/details/83387623

版权

本文详细介绍了如何使用Python的sklearn库实现KNN（K近邻）和KD树算法，包括步骤和关键代码，是机器学习初学者的实用教程。

摘要由CSDN通过智能技术生成

python+sklearn实现KNN及KD树算法

from sklearn import datasets# 导入内置数据集模块
from sklearn.neighbors import KNeighborsClassifier# 导入sklearn.neighbors模块中KNN类
import numpy as np
from sklearn.neighbors import KDTree#导入KD树类

np.random.seed(0)# 设置随机种子，不设置的话默认是按系统时间作为参数，因此每次调用随机模块时产生的随机数都不一样设置后每次产生的一样
iris = datasets.load_iris()# 导入鸢尾花的数据集，iris是一个类似于结构体的东西，内部有样本数据，如果是监督学习还有标签数据
iris_x = iris.data# 样本数据150*4二维数据，代表150个样本，每个样本4个属性分别为花瓣和花萼的长、宽
iris_y = iris.target# 长150的以为数组，样本数据的标签
indices = np.random.permutation(len(iris_x)) # permutation接收一个数作为参数(150),产生一个0-149一维数组，只不过是随机打乱的，当然她也可以接收一个一维数组作为参数，结果是直接对这个数组打乱
iris_x_train = iris_x[indices[:-10]]# 随机选取140个样本作为训练数据集
iris_y_train = iris_y[indices[:-10]]# 并且选取这140个样本的标签作为训练数据集的标签
iris_x_test = iris_x[indices[-10:]]# 剩下的10个样本作为测试数据集
iris_y_test = iris_y[indices[-10:]]# 并且把剩下10个样本对应标签作为测试数据及的标签

knn

最低0.47元/天解锁文章