1 Scikit-learn工具介绍

• 目前稳定版本0.19.1

1.1 安装

pip3 install scikit-learn==0.19.1

查看是否安装成功

import sklearn

注：安装scikit-learn需要Numpy, Scipy等库

1.2 Scikit-learn包含的内容

• 分类、聚类、回归
• 特征工程
• 模型选择、调优

2 K-近邻算法API

• sklearn.neighbors.KNeighborsClassifier(n_neighbors=5)
  • n_neighbors：int,可选（默认= 5),查询默认使用的邻居数

3 案例

• 导入模块

from sklearn.neighbors import KNeighborsClassifier

• 构造数据集

x = [[0], [1], [2], [3]]
y = [0, 0, 1, 1]

• 机器学习 -- 模型训练

# 实例化API
estimator = KNeighborsClassifier(n_neighbors=2)
# 使用fit方法进行训练
estimator.fit(x, y)

estimator.predict([[1]])

标签：neighbors,fit,KNeighborsClassifier,近邻,estimator,初步,api,learn,sklearn
来源： https://www.cnblogs.com/yuyingblogs/p/15262982.html