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

网友评论