KNN 有监督分类算法,根据距离相近的邻居类别,来判定自己的类别。
from sklearn.neighbors import KNeighborsClassifier
clf = KNeighborsClassifier()
clf.fit(features_train, labels_train)
pre_test = clf.predict(features_test)
PS:自用的 不做解释 作者:一个吃货帅锅
来源:oschina
链接:https://my.oschina.net/u/2687320/blog/1618238