03 scikit-learn中的KNN

KNN详解 https://scikit-learn.org/stable/modules/generated/sklearn.neighbors.KNeighborsClassifier.html

from sklearn.neighbors import KNeighborsClassifier
kNN_classifier = KNeighborsClassifier(n_neighbors=6)
kNN_classifier.fit(X_train, y_train)  # 建模
X_predict = x.reshape(1, -1)
kNN_classifier.predict(X_predict)  # array([1])
y_predict = kNN_classifier.predict(X_predict)
print(y_predict[0])  # 1