机器学习框架sklearn之K近邻算法
创始人
2025-05-30 01:01:44
0

原理

K Nearest Neighbor算法又叫KNN算法,这个算法是机器学习里面一个比较经典的算法

  • 定义

    如果一个样本在特征空间中的K个最相似(即特征空间中最邻近)的样本大多数属于某一个类别,则该样本也属于这个类别。

  • 距离公式

    两个样本的距离可以通过如下公式计算,又叫欧氏距离
    比如说,点a(a1,a2,a3)和点b(b1,b2,b3)的距离为:
    在这里插入图片描述
    其他的距离公式:

    曼哈顿距离:绝对值距离

注:

K值取的过小,容易受到异常点的影响

K值取的过大,容易受到样本不均衡的影响

API

sklearn.neighbors.KNeighborsClassifier(n_neighbors=5,algorithm='auto')-n_neighbors:int,可选(默认=5),n_neighbors查询默认使用的邻居数
-algorithm:{'auto','ball_tree','kd_tree','brute'},可选用于计算最近邻居的算法:'ball_tree'将会使用BallTree,'kd_tree'将使用KDTree,'auto'将尝试根据传递给fit方法的值来决定最合适的算法

使用

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier
def knn_demo():#1.获取数据iris=load_iris()#2.划分数据集x_train,x_test,y_train,y_test=train_test_split(iris.data,iris.target,random_state=6)#3.特征工程:标准化transfer=StandardScaler()x_train=transfer.fit_transform(x_train)x_test=transfer.transform(x_test)#4.KNN算法预估器estimator=KNeighborsClassifier(n_neighbors=3)estimator.fit(x_train,y_train)# 5.评估模型#方法一y_predict=estimator.predict(x_test)print("直接比对真实值和预测值:\n",y_test==y_predict)# 方法二score=estimator.score(x_test,y_test)print("准确率为:\n",score)return None

总结

  • 优点

    简单、易于理解、易于实现、无需训练

  • 缺点

    懒惰算法,对测试样本分类时的计算量大,内存开销大

    必须指定K值,K值选择不当则分类精度不能保证

  • 使用场景

    小数据场景,几千~几万样本,具体场景具体业务去测试

相关内容

热门资讯

公共政策正向排斥:理论建构与中... 党的二十届四中全会强调,“十五五”时期经济社会发展必须“尊重人民主体地位,紧紧依靠人民,维护人民根本...
纪法百科•一图读懂应知应会党纪... 来源:中央纪委国家监委网站
扶持政策升级 近百家短剧产业链... 人民网北京11月23日电 (记者董兆瑞)11月22日,中国影都短剧产业发展一周年成果发布会召开。记者...
每周股票复盘:佛山照明(000... 截至2025年11月21日收盘,佛山照明(000541)报收于6.08元,较上周的6.52元下跌6....
最严“保送生”政策出台 “豪赌... 羊城晚报记者 蒋隽 近日,五大学科竞赛的全国比赛成绩陆续揭晓,各大高中频传获奖、保送清北、破格入围强...
因劳动争议,张文国起诉三维股份 天眼查APP显示,近日,张文国新增一则开庭公告,案由为“劳动争议”,原告为张文国,被告为三维控股集团...
“免费梅州”政策 持续助力招好... ●南方日报记者 黄培强 唐林珍 “此次项目的成功落地将进一步壮大园区汽车零部件产业集群,助力梅州产业...
四部门联合发文规范实施金融惠企... 原标题:四部门联合发文规范实施金融惠企财政贴息政策 从四方面明确贴息业务合规操作禁区 记者11月21...
法规延期!利好中国轮胎业 近日, 欧盟理事会通过欧盟零毁林法案(EUDR)修订授权,正式启动与欧洲议会的谈判进程。 此次修订...
欧盟MDR法规下的CE认证:医... 欧盟的医疗器械MDR(Medical Device Regulation,医疗器械法规)是自2021...