机器学习------knn 之二------API使用之SCIKIT-LEARN初探

机器学习------knn
二、API使用之SCIKIT-LEARN初探
1.SCIKIT-LEARN
sklearn的优势:
文档多,且规范
包含的算法多
实现起来容易
knn中的api
sklearn.neighbors.KNeighborsClassifier(n_neighbors=5)

 1.1安装

pip3 install scikit-learn==0.19.1

2.包含内容：
分类、聚类、回归
特征工程
模型选择、调优

3.使用

K-近邻算法API
sklearn.neighbors.KNeighborsClassifier(n_neighbors=5)
n_neighbors：int,可选（默认= 5），k_neighbors查询默认使用的邻居数

4.代码：

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)

estimator.predict([[1]])

n_neighbors=1
不写默认5，样本数少于5会报错