6.3.2KNN最邻近分类的python实现方法neighbor

2018-06-01  本文已影响0人  夏日春风

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--创建散点图

上一篇 下一篇

猜你喜欢

热点阅读