From e727c0d9ab2e40071a5c846d2753a85b82291a75 Mon Sep 17 00:00:00 2001
From: lzh <202210423221@qq.com>
Date: Wed, 16 Oct 2024 23:59:19 +0800
Subject: [PATCH] 2

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

diff --git a/function.py b/function.py
index 333e89a..22ecd9e 100644
--- a/function.py
+++ b/function.py
@@ -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)