#导入模块
from sklearn.neighbors import KNeighborsClassifier
#构造数据集
x = [[0],[1],[2],[3]]
y = [0,0,1,1]
#实例化API
estimator = KNeighborsClassifier(n_neighbors = 1)
#使用fit方法进行训练
estimator.fit(x, y)
#模型评估
ret = estimator.predict([[1]])
print(ret)
结果:[0]
说明:
- sklearn的优势:
- 文档多,且规范
- 包含的算法多
- 实现起来容易
- knn中的api
- sklearn.neighbors.KNeighborsClassifier(n_neighbors=5)