From a9262600987dc62a8d8de79b11e1faddaceb9a12 Mon Sep 17 00:00:00 2001
From: fox <3147253307@qq.com>
Date: Wed, 23 Oct 2024 11:02:12 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/function.py b/function.py
index 60936bb..a60e089 100644
--- a/function.py
+++ b/function.py
@@ -63,8 +63,8 @@ def classify(input_vector, dataSet, labels, k=3):
     # 按距离升序排列，选出距离最小的k个点的索引
     nearest_neighbors = distances.argsort()[:k]
 
-    # 获取这k个邻居的标签
-    neighbor_labels = [labels[i] for i in nearest_neighbors]
+    # 获取这k个邻居的标签，并转为列表
+    neighbor_labels = list(labels[nearest_neighbors])
 
     # 返回出现次数最多的标签
     predicted_label = max(set(neighbor_labels), key=neighbor_labels.count)