jiarui_li/DeepHealth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename age_encoder parameter to age_encoder_type for clarity in Trainer class

Browse Source
This commit is contained in:
Jiarui Li
2026-01-08 11:34:44 +08:00
parent 8293f7ee24
commit 1d1f568a3f
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
train.py
View File

@@ -178,7 +178,7 @@ class Trainer:
n_head=cfg.n_head, n_head=cfg.n_head,
n_layer=cfg.n_layer, n_layer=cfg.n_layer,
pdrop=cfg.pdrop, pdrop=cfg.pdrop,
age_encoder=cfg.age_encoder, age_encoder_type=cfg.age_encoder,
n_dim=n_dim, n_dim=n_dim,
n_cont=dataset.n_cont, n_cont=dataset.n_cont,
n_cate=dataset.n_cate, n_cate=dataset.n_cate,
@@ -190,7 +190,7 @@ class Trainer:
n_head=cfg.n_head, n_head=cfg.n_head,
n_layer=cfg.n_layer, n_layer=cfg.n_layer,
pdrop=cfg.pdrop, pdrop=cfg.pdrop,
age_encoder=cfg.age_encoder, age_encoder_type=cfg.age_encoder,
n_dim=n_dim, n_dim=n_dim,
n_cont=dataset.n_cont, n_cont=dataset.n_cont,
n_cate=dataset.n_cate, n_cate=dataset.n_cate,
@@ -212,7 +212,7 @@ class Trainer:
n_head=cfg.n_head, n_head=cfg.n_head,
n_layer=cfg.n_layer, n_layer=cfg.n_layer,
pdrop=cfg.pdrop, pdrop=cfg.pdrop,
age_encoder=cfg.age_encoder, age_encoder_type=cfg.age_encoder,
n_dim=n_dim, n_dim=n_dim,
n_cont=dataset.n_cont, n_cont=dataset.n_cont,
n_cate=dataset.n_cate, n_cate=dataset.n_cate,
@@ -224,7 +224,7 @@ class Trainer:
n_head=cfg.n_head, n_head=cfg.n_head,
n_layer=cfg.n_layer, n_layer=cfg.n_layer,
pdrop=cfg.pdrop, pdrop=cfg.pdrop,
age_encoder=cfg.age_encoder, age_encoder_type=cfg.age_encoder,
n_dim=n_dim, n_dim=n_dim,
n_cont=dataset.n_cont, n_cont=dataset.n_cont,
n_cate=dataset.n_cate, n_cate=dataset.n_cate,