diff --git a/function.py b/function.py
index 4475b60..7e4a279 100644
--- a/function.py
+++ b/function.py
@@ -70,8 +70,8 @@ def predict_svm(model, scaler, X):
 
 if __name__ == "__main__":
     # 加载数据
-    X_train, y_train = load_data("./horse-colic.data")
-    X_test, y_test = load_data("./horse-colic.test")
+    X_train, y_train = load_data("./data/horse-colic.data")
+    X_test, y_test = load_data("./data/horse-colic.test")
     
     # 训练SVM
     model, scaler = train_svm(X_train, y_train)