From efbcb0589762fdb842d0fe62aabb0320518312ff Mon Sep 17 00:00:00 2001
From: cy <939601402@qq.com>
Date: Sat, 26 Oct 2024 19:37:47 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/function.py b/function.py
index 0f1346c..7efd0a1 100644
--- a/function.py
+++ b/function.py
@@ -137,8 +137,8 @@ if __name__ == "__main__":
     X_test, y_test = load_data("./horse-colic.test")
 
     # 训练对率回归模型
-    weights = train_logreg(X_train, y_train, learning_rate=0.2, iterations=200, batch_size=32, regularization='L1',
-                           lambda_reg=0.15)
+    weights = train_logreg(X_train, y_train, learning_rate=0.25, iterations=1500, batch_size=16, regularization='L2',
+                           lambda_reg=0.5)
     # 计算并输出准确率
     acc = accuracy(X_test, y_test, weights)
     print(f"模型的准确率: {acc:.2f}")