更新 function.py

2024-10-26 20:23:01 +08:00
parent 44d0338772
commit 1423da5ab5
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.05, iterations=1500, batch_size=64, regularization='L2', lambda_reg=0.05)
+    final_weights = train_logreg(X_train, y_train, learning_rate=0.01, iterations=1000, batch_size=16, regularization='L1', lambda_reg=0.01)

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