更新 function.py

2024-10-26 20:28:35 +08:00
parent 1423da5ab5
commit 3dc01d2809
1 changed files with 1 additions and 1 deletions
@@ -129,7 +129,7 @@ if __name__ == "__main__":
    X_train, y_train = load_data("./data/horse-colic.data")
    X_test, y_test = load_data("./data/horse-colic.test")

-    final_weights = train_logreg(X_train, y_train, learning_rate=0.01, iterations=1000, batch_size=16, regularization='L1', lambda_reg=0.01)
+    final_weights = train_logreg(X_train, y_train, learning_rate=0.05, iterations=1000, batch_size=16, regularization='L1', lambda_reg=0.05)

    # 计算并输出最终测试集的准确率
    final_accuracy = accuracy(X_test, y_test, final_weights)