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]
机器学习:KNN对iris分类
猜你喜欢
转载自blog.csdn.net/Benanan/article/details/84062374
今日推荐
周排行