6.3.2KNN最邻近分类的python实现方法neighbors模块

1--from sklearn import neighbors #导入KNN分类模块

导入模块,创建数据

2--设置KNN分类器,带入数据进行训练

knn = neighbors.KNeighborsClassifier()#取得KNN分类器

knn.fit(data[['fight','kiss']],data['type'])#进行训练

knn.predict('fight','kiss'的2维数组)

3--根据创建的数据,用训练后KNN模型来进行贴标签

data2['type'] = knn.predict(data2)

4--创建散点图

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容