更新 function.py

function.py

@@ -29,7 +29,7 @@ id = "202210423107"
 name = "刘燚燊"
 
 
-def load_data(directory):
+def load_dataset(directory):
     """
     读取训练集或测试集的数字图像，转换为矩阵格式。
     Args:
@@ -81,3 +81,13 @@ def classify(input_vector, dataSet, labels, k=3):
     most_common_label = Counter(k_neighbors_labels).most_common(1)[0][0]
     
     return most_common_label
+    # 获取距离最近的 k 个样本的索引
+    k_neighbors_indices = distances.argsort()[:k]
+    
+    # 获取这些样本对应的标签
+    k_neighbors_labels = labels[k_neighbors_indices]
+    
+    # 使用投票机制决定分类
+    most_common_label = Counter(k_neighbors_labels).most_common(1)[0][0]
+    
+    return most_common_label