更新 function.py

function.py

@@ -98,9 +98,10 @@ def img_to_vector(filename):
 
 
 if __name__ == "__main__":
-    # 测试代码
+    # 加载训练数据
     train_data, train_labels = load_dataset("trainingDigits")
 
+    # 测试数据
     test_data = (
         "00000000000000110000000000000000"
         "00000000000011111000000000000000"
@@ -138,6 +139,6 @@ if __name__ == "__main__":
 
     test_vector = np.array([int(char) for char in test_data]).reshape(1, 1024)
 
-    # 使用默认的欧氏距离和不加权的投票机制
-    predicted_label = classify(test_vector, train_data, train_labels)
-    print(f"预测的标签（普通分类）: {predicted_label}")
+    # 使用曼哈顿距离进行分类
+    predicted_label = classify(test_vector, train_data, train_labels, distance_metric='manhattan')
+    print(f"预测的标签（曼哈顿距离）: {predicted_label}")