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

Remove legacy event and mixed distribution paths

Browse Source
This commit is contained in:
Jiarui Li
2026-08-01 14:23:18 +08:00
parent dfb22adf2d
commit de6f9b75b9
22 changed files with 370 additions and 463 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
targets.py
View File

@@ -6,7 +6,7 @@ import numpy as np
PAD_IDX = 0
CHECKUP_IDX = 1
RESERVED_IDX = 1
NO_EVENT_IDX = 2
DAYS_PER_YEAR = 365.25
@@ -105,7 +105,7 @@ def build_next_token_targets(
target_events: [x1, x2, ..., xN-1]
target_times_years: [t1, t2, ..., tN-1] / 365.25
This function does not ignore PAD/CHECKUP/NO_EVENT. Ignoring belongs to
This function does not ignore PAD/RESERVED/NO_EVENT. Ignoring belongs to
the loss function because different objectives may use different ignore ids.
"""
labels = _as_numpy_1d(labels, "labels", np.int64)