2. k近邻算法api初步使用
Last updated
Last updated
pip3 install scikit-learn==0.19.1import sklearnfrom sklearn.neighbors import KNeighborsClassifierx = [[0], [1], [2], [3]]
y = [0, 0, 1, 1]# 实例化API
estimator = KNeighborsClassifier(n_neighbors=2)
# 使用fit方法进行训练
estimator.fit(x, y)
estimator.predict([[1]])