更新 function.py

2024-10-21 16:34:09 +08:00
parent 4d57375554
commit ef9fdec4ff
1 changed files with 3 additions and 3 deletions
@@ -134,9 +134,9 @@ def img_to_vector(filename):

 if __name__ == "__main__":
    # 测试代码
-   train_data, train_labels = load_dataset("trainingDigits")
+    train_data, train_labels = load_dataset("trainingDigits")

-   test_data = (
+    test_data = (
        "00000000000000110000000000000000"
        "00000000000011111000000000000000"
        "00000000000011111100000000000000"
@@ -179,4 +179,4 @@ if __name__ == "__main__":
    
    # 使用 KDTree 加速的 KNN 进行预测
    predicted_label_kdtree = classify(test_vector, train_data, train_labels, k=5, use_kdtree=True)
-    print(f"KDTree 加速后的预测标签: {predicted_label_kdtree}")
+    print(f"KDTree 加速后的预测标签: {predicted_label_kdtree}")