3 # -*- coding: utf-8 -*-
4 from sklearn
import neighbors
5 from sklearn
.datasets
import load_iris
11 colors
= ["red", "green", "blue"]
14 clf
= neighbors
.KNeighborsClassifier(nb_voisins
)
16 print("kNN prediction on [5.4, 3.2, 1.6, 0.4]:")
17 print(clf
.predict([[5.4, 3.2, 1.6, 0.4]]))
18 print("kNN probability prediction on [5.4, 3.2, 1.6, 0.4]:")
19 print(clf
.predict_proba([[5.4, 3.2, 1.6, 0.4]]))
20 print("kNN score on Iris data:")
21 print(clf
.score(X
, Y
))
22 print("kNN prediction error(s) on Iris data:")