使用KNN进行图书推荐

import numpy as np
from sklearn import neighbors
knn = neighbors.KNeighborsClassifier(1)   #取得 knn 分类器
data = np.array([[1.1, 1.5, 1.4, 0.2],
                 [1.9, 1.0, 1.4 ,0.2],
                 [1.7, 1.2, 1.3, 0.2],
                 [2.6, 2.1, 1.5, 0.2],
                 [2.0, 2.6, 1.4, 0.2]])
labels = np.array(['A','B','C','D','E'])
knn.fit(data,labels)
print("预测结果：",knn.predict(np.array([[1.6, 1.5, 1.2, 0.1]]).reshape(1,-1)))