更新 function.py

2024-10-31 11:44:13 +08:00
parent b9cd65c9f2
commit 2217c49cac
1 changed files with 2 additions and 2 deletions
@@ -83,10 +83,10 @@ def predict_logreg(inX, weights):
 if __name__ == "__main__":
    # 以下代码可用于测试，可自行随意修改
    # 加载训练集
-    X_train, y_train = load_data("./data/horse-colic.data")
+    X_train, y_train = load_data("/data/horse-colic.data")

    # 加载测试集
-    X_test, y_test = load_data("./data/horse-colic.test")
+    X_test, y_test = load_data("/data/horse-colic.test")

    # 训练对率回归模型
    weights = train_logreg(X_train, y_train, learning_rate=0.01, iterations=1000)