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)