From 26098f5aaf2d4d1f70618084d2fbb80acd417bae Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=88=92=E9=85=92=E7=9A=84=E6=9D=8E=E7=99=BD?=
 <670939375@qq.com>
Date: Wed, 2 Apr 2025 20:37:28 +0800
Subject: [PATCH] Fix2: Provide a seed for the random_state parameter.

---
 model_pro/LSTM_model.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/model_pro/LSTM_model.py b/model_pro/LSTM_model.py
index 7ec84d7..1a46b6f 100644
--- a/model_pro/LSTM_model.py
+++ b/model_pro/LSTM_model.py
@@ -308,6 +308,10 @@ class LSTMModelManager:
         return loss + self.alpha * loss_adv
     
     def train_logistic_regression(self, train_texts, train_labels, val_texts=None, val_labels=None):
+        """训练逻辑回归基线模型"""
+        # 设置随机种子以确保可重现性
+        np.random.seed(self.random_seed)
+        
         vectorizer = TfidfVectorizer(max_features=5000)
         X_train = vectorizer.fit_transform(train_texts)
         
@@ -323,7 +327,8 @@ class LSTMModelManager:
         
         lr_model = LogisticRegression(
             class_weight='balanced',
-            random_state=self.random_seed  # 添加随机种子
+            random_state=self.random_seed,  # 添加随机种子
+            max_iter=1000  # 增加最大迭代次数以确保收敛
         )
         lr_model.fit(X_train, y_train)