更新function.py

2024-10-23 11:02:12 +08:00
parent 14f011c9d5
commit a926260098
1 changed files with 2 additions and 2 deletions
@@ -63,8 +63,8 @@ def classify(input_vector, dataSet, labels, k=3):
    # 按距离升序排列，选出距离最小的k个点的索引
    nearest_neighbors = distances.argsort()[:k]
-    # 获取这k个邻居的标签
+    # 获取这k个邻居的标签，并转为列表
-    neighbor_labels = [labels[i] for i in nearest_neighbors]
+    neighbor_labels = list(labels[nearest_neighbors])
    # 返回出现次数最多的标签
    predicted_label = max(set(neighbor_labels), key=neighbor_labels.count)