从 scikit KNeighborsClassifier 打印最近邻居的标签？

Question

我正在使用 KNeighborsClassifier 算法来训练我的数据，如下所示：

knn_clf = neighbors.KNeighborsClassifier(n_neighbors=3, 
algorithm="ball_tree", weights='distance')

我从以下位置获取数据的最近 3 个邻居：

closest_distances = knn_clf.kneighbors(faces_encodings, n_neighbors=3)
print(closest_distances) #printouts: (array([[0.1123 , 0.29189484, 0.312]]

我想将这 3 个最近邻居的距离与标签联系起来。有任何想法吗 ？

我的训练数据：X:[0.1,0.2,0.3,0.4,0.5] y:[一、二、三、四、五]。如果我给出预测（0.2），我如何将找到的距离的标签链接为二、三、四？

谢谢，

Answer 1

您不应该像您那样将knn_clf.kneighbors 分配给单个变量closest_distances； kneighbors方法returns two variables：

返回：

dist : 数组

表示点长度的数组，仅在 return_distance=True 时出现

ind : 数组

人口矩阵中最近点的索引。

所以，你应该给

closest_distances, indices = knn_clf.kneighbors(faces_encodings, n_neighbors=3)

indices 变量将包含所需的索引。