Enhance DeepHealth model to incorporate CHECKUP state tokens in next-step training and evaluation, update dataset cache versioning, and improve handling of observed event histories.

2026-06-15 14:10:09 +08:00
parent 593ecd2e71
commit c3e49db859
8 changed files with 111 additions and 86 deletions
--- a/eval_data.py
+++ b/eval_data.py
@@ -14,9 +14,9 @@ class AllFutureSequenceEvalDataset:
    """
    Eval-only sequence view for all-future checkpoints.

-    All-future training uses pure disease histories, so token-level and landmark
-    evaluation should not reuse the next-step dataset view that contains
-    imputed <NO_EVENT> gap tokens.
+    All-future training uses the observed history, including CHECKUP state
+    tokens, without reusing the next-step view that contains imputed
+    <NO_EVENT> gap tokens.
    """

    def __init__(