2

2024-10-16 23:59:19 +08:00
parent 8719f435de
commit e727c0d9ab
1 changed files with 2 additions and 2 deletions
@@ -97,8 +97,8 @@ def classify(input_vector, dataSet, labels, k=3):
    return label_count.most_common(1)[0][0]

 if __name__ == "__main__":
-    train_dir = "trainingDigits"  # 训练集目录
-    test_dir = "testDigits"  # 测试集目录
+    train_dir = "./trainingDigits"  # 训练集目录
+    test_dir = "./testDigits"  # 测试集目录
    
    train_data, train_labels = load_dataset(train_dir)
    test_data, test_labels = load_dataset(test_dir)