更新 function.py

2024-10-31 17:11:12 +08:00
parent 83e0587902
commit 1afe2c29c7
1 changed files with 2 additions and 0 deletions
@@ -100,9 +100,11 @@ if __name__ == "__main__":
    # 以下代码可用于测试，可自行随意修改
    # 加载训练集
    X_train, y_train = load_data("./data/horse-colic.data")
+    X_train = preprocess_data(X_train)

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

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