更新 function.py

2024-11-14 17:07:57 +08:00
parent e6527b297b
commit 65926187ab
1 changed files with 2 additions and 2 deletions
@@ -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)