KNeighborsClassifier
Классификатор по k ближайшим соседям.
Сигнатура
KNeighborsClassifier(n_neighbors=5, weights='uniform')Относит объект к классу, преобладающему среди его n_neighbors ближайших соседей. Чувствителен к масштабу признаков, поэтому их стоит стандартизировать.
from sklearn.neighbors import KNeighborsClassifier
clf = KNeighborsClassifier(n_neighbors=7)
clf.fit(X_train, y_train)
y_pred = clf.predict(X_test)