JiaruiLi/DeepHealthExpo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Expose TimesNet internal dimension

Browse Source
This commit is contained in:
Jiarui Li
2026-07-09 16:22:34 +08:00
parent 8078fcb835
commit 552e09aa01
3 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
evaluate_auc.py
View File

@@ -324,7 +324,7 @@ def build_model_from_dataset(args: argparse.Namespace, cfg: Dict[str, Any], data
dist_mode=str(cfg_get(args, cfg, "dist_mode", "exponential")),
dropout=float(cfg_get(args, cfg, "dropout", 0.0)),
use_exposure_encoder=bool(cfg_get(args, cfg, "use_exposure_encoder", False)),
exposure_d_model=cfg_get(args, cfg, "exposure_d_model", 64),
exposure_d_model=cfg_get(args, cfg, "d_model", 64),
exposure_n_layers=int(cfg_get(args, cfg, "exposure_n_layers", 2)),
exposure_top_k=int(cfg_get(args, cfg, "exposure_top_k", 2)),
exposure_n_backbone_blocks=int(cfg_get(args, cfg, "exposure_n_backbone_blocks", 1)),