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

evaluate_auc: adapt to new utils.load_model signature (infer checkpoint from config)

Browse Source
This commit is contained in:
Jiarui Li
2025-10-22 11:40:46 +08:00
parent 6801e5bdbb
commit 93cf2018d2
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
evaluate_auc.py
View File

@@ -452,7 +452,6 @@ def main():
# Load model checkpoint and initialize model. # Load model checkpoint and initialize model.
model = load_model( model = load_model(
config_path=f'config_{model_name}.json', config_path=f'config_{model_name}.json',
model_path=f'best_model_{model_name}.pt',
device=device, device=device,
) )
model.eval() model.eval()