使用sklearn中的kNN分类大概有如下几个步骤:
1. 需要使用sklearn.neighbors包中的KNeighborsClassifier类,需要进行包和类的导入。
from sklearn.neighbors import KNeighborsClassifier
import numpy as np
2. 实例化KNeighborsClassifier对象。
'''
KNeighborsClassifier(
n_neighbors=5,
weights='uniform',
algorithm='auto',
leaf_size=30,
p=2,
metric='minkowski',
metric_params=None,
n_jobs=None,
**kwargs,
)
Docstring: Classifier implementing the k-nearest neighbors vote.
'''
kNN_classifier = KNeighborsClassifier(n_neighbors=6)
3. 对KNeighborsClassifier对象传入训练集进行训练。
'''
Signature: kNN_classifier.fit(X, y)
Docstring:
Fit the m