From a1dfa2ad3062c7a4653a84dd410462ccda7697e9 Mon Sep 17 00:00:00 2001
From: dong <2938690875@qq.com>
Date: Sat, 2 Nov 2024 13:55:53 +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 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/function.py b/function.py
index 8f9327f..bde0b00 100644
--- a/function.py
+++ b/function.py
@@ -92,7 +92,7 @@ if __name__ == "__main__":
     X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
 
     # 训练逻辑回归模型，设置 batch size
-    batch_size = 32
+    batch_size = 64
     weights = train_logreg(X_train, y_train, learning_rate=0.1, iterations=1000, lambda_reg=0.01, batch_size=batch_size)
 
     # 测试单个数据