RYDE-WORK/lnp_ml
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/lnp_ml.git synced 2026-03-21 09:36:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加超参搜索范围

Browse Source
This commit is contained in:
RYDE-WORK 2026-02-28 13:57:31 +08:00
parent 9e4800b77e
commit 48de4d5d37
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lnp_ml/modeling/final_train_optuna_cv.py
View File

@ -282,8 +282,8 @@ def run_optuna_cv(
# 搜索训练超参数 # 搜索训练超参数
dropout = trial.suggest_float("dropout", 0.1, 0.5) dropout = trial.suggest_float("dropout", 0.1, 0.5)
lr = trial.suggest_float("lr", 1e-5, 3e-4, log=True) lr = trial.suggest_float("lr", 1e-5, 1e-3, log=True)
weight_decay = trial.suggest_float("weight_decay", 1e-5, 1e-3, log=True) weight_decay = trial.suggest_float("weight_decay", 1e-5, 1e-1, log=True)
backbone_lr_ratio = trial.suggest_float("backbone_lr_ratio", 0.01, 1.0, log=True) backbone_lr_ratio = trial.suggest_float("backbone_lr_ratio", 0.01, 1.0, log=True)
# 3-fold CV # 3-fold CV

4
lnp_ml/modeling/nested_cv_optuna.py
View File

@ -395,8 +395,8 @@ def run_inner_optuna(
# 搜索训练超参数 # 搜索训练超参数
dropout = trial.suggest_float("dropout", 0.1, 0.5) dropout = trial.suggest_float("dropout", 0.1, 0.5)
lr = trial.suggest_float("lr", 1e-5, 3e-4, log=True) lr = trial.suggest_float("lr", 1e-5, 1e-3, log=True)
weight_decay = trial.suggest_float("weight_decay", 1e-5, 1e-3, log=True) weight_decay = trial.suggest_float("weight_decay", 1e-5, 1e-1, log=True)
backbone_lr_ratio = trial.suggest_float("backbone_lr_ratio", 0.01, 1.0, log=True) backbone_lr_ratio = trial.suggest_float("backbone_lr_ratio", 0.01, 1.0, log=True)
# 内层 3-fold CV # 内层 3-fold CV