diff --git a/function.py b/function.py
index 404b97b..2525668 100644
--- a/function.py
+++ b/function.py
@@ -92,8 +92,9 @@ def load_data(filepath):
 
 # main函数，程序的入口点
 def main():
-    X_train, y_train = load_data("./data/horse-colic.data")  # 加载训练数据
-    X_test, y_test = load_data("./data/horse-colic.test")  # 加载测试数据
+    # 假设data文件夹和function.py在同一目录下
+    X_train, y_train = load_data("./data/horse-colic.data")
+    X_test, y_test = load_data("./data/horse-colic.test")
 
     if X_train is not None and X_test is not None:
         # 特征缩放