Toggle navigation
Toggle navigation
This project
Loading...
Sign in
万朱浩
/
Venue-Ops
Go to a project
Toggle navigation
Projects
Groups
Snippets
Help
Toggle navigation pinning
Project
Activity
Repository
Pipelines
Graphs
Issues
0
Merge Requests
0
Wiki
Network
Create a new issue
Builds
Commits
Authored by
戒酒的李白
2025-04-02 20:37:28 +0800
Browse Files
Options
Browse Files
Download
Email Patches
Plain Diff
Commit
26098f5aaf2d4d1f70618084d2fbb80acd417bae
26098f5a
1 parent
296d8e4b
Fix2: Provide a seed for the random_state parameter.
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
1 deletions
model_pro/LSTM_model.py
model_pro/LSTM_model.py
View file @
26098f5
...
...
@@ -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
)
...
...
Please
register
or
login
to post a comment