knn算法 k近邻算法demo(1)

from sklearn.neighbors import KNeighborsClassifier
import numpy as np

KNN_classifier=KNeighborsClassifier(n_neighbors=2)
X=np.arange(10).reshape(5,2)

print(X)
y=[0,0,0,1,1]
KNN_classifier.fit(X,y)
x=np.array([[3,2]])
result=KNN_classifier.predict(x)
print(result)