从 sklearn.neighbors 导入 KNeighborsClassifier

from sklearn.neighbors import KNeighborsClassifier

定义训练集

X = [[0,7], [1,5], [2,4], [3,1], [4,3], [5,6], [6,7], [7,9], [8,11]] y = [0, 0, 0, 1, 1, 1, 2, 2, 2]

任务1：创建 KNeighborsClassifier 对象，令 k=5

########## Begin ########## kNN = KNeighborsClassifier(n_neighbors=5) ########## End ##########

任务2：调用 fit 函数执行训练过程

########## Begin ########## kNN.fit(X, y) ########## End ##########

调用 predict 函数进行预测

pre_y = kNN.predict([[1,1], [7,5], [8,5]])

打印结果

print(pre_y