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

Fix calibration death token lookup

Browse Source
This commit is contained in:
Jiarui Li
2026-08-03 08:33:21 +08:00
parent 4f102fb271
commit 271a678e73
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
evaluate_calibration.py
View File

@@ -1755,7 +1755,7 @@ def main() -> None:
if landmark_ages.size == 0: if landmark_ages.size == 0:
raise ValueError("No landmark ages selected.") raise ValueError("No landmark ages selected.")
death_token_ids = _get_death_token_ids(dataset, labels_meta) death_token_ids = _get_death_token_ids(dataset)
landmark_dataset = LandmarkDataset( landmark_dataset = LandmarkDataset(
dataset=dataset, dataset=dataset,
subset_indices=subset_indices, subset_indices=subset_indices,