机器学习:KNN对iris分类

from sklearn import neighbors #调用neighbors
from sklearn import datasets #调用datasets里面的iris数据

knn = neighbors.KNeighborsClassifier()#knn分类器

#调用数据集
iris = datasets.load_iris()
#print(iris)

#使用fit建立模型
knn.fit(iris.data,iris.target)
#开始预测
a = ([[0.1,0.2,0.3,0.4]])
c = ([[0.8,4.9,4.3,1.9]])
b = ([[4.3,2.6,3.8,6.8]])

predict_a = knn.predict(a)
predict_b = knn.predict(b)
predict_c = knn.predict(c)

print("predict_a\n",predict_a,
      "\npredict_b\n",predict_b,
      "\npredict_c\n",predict_c)

#predict_a
# [0] 
#predict_b
# [2] 
#predict_c
# [1]

猜你喜欢

转载自blog.csdn.net/Benanan/article/details/84062374