d3e68ac09b
Avoid overflow in attribution variance
2026-07-01 10:58:09 +08:00
16df74c04d
Skip attribution for next-token runs
2026-07-01 10:47:09 +08:00
d9dc115e09
Report disease parameter attribution
2026-07-01 10:16:51 +08:00
3cf756ccb0
Add landmark attribution dependencies
2026-07-01 09:55:38 +08:00
93450ab06b
Update missing evaluation runner
2026-07-01 09:19:42 +08:00
84ae07bf48
Add missing training runs script
2026-06-29 15:40:55 +08:00
ca294193f4
Clarify empty extra-info logging
2026-06-29 11:26:48 +08:00
d4a29063d6
Add no extra-info type list
2026-06-29 11:14:50 +08:00
00ccedf82f
Report death distribution parameters for disease attribution
2026-06-29 10:42:32 +08:00
fe826c4052
Defer attribution output until after GPU pass
2026-06-27 16:11:49 +08:00
980dd7c282
Align disease attribution batching with survival evaluator
2026-06-27 16:01:35 +08:00
517d573d3b
Reduce attribution GPU synchronization
2026-06-27 15:56:12 +08:00
20686c128f
Vectorize per-disease attribution batches
2026-06-27 15:51:12 +08:00
8928688aca
Optimize per-disease attribution scanning
2026-06-27 15:43:27 +08:00
8ef88ff325
Pad attribution batches before concatenation
2026-06-27 15:38:32 +08:00
a7e969ae51
Summarize disease mortality attribution by age and sex
2026-06-27 15:33:50 +08:00
34353d33d5
Scan all diseases for mortality attribution
2026-06-27 15:30:38 +08:00
128a8ce655
Add single-disease mortality attribution script
2026-06-27 15:24:15 +08:00
01629cec56
Fix mortality attribution active rows
2026-06-27 14:29:32 +08:00
2ae92da0ec
Stream mortality attribution batches
2026-06-27 14:08:02 +08:00
12265bd248
Batch mortality attribution queries
2026-06-27 13:57:16 +08:00
0b7c866292
Add mortality attribution evaluation
2026-06-27 13:34:30 +08:00
00de2ff4a9
Fix event-free survival inference tensor use
2026-06-27 12:08:19 +08:00
307704f74d
Simplify event-free survival data loading
2026-06-27 12:02:48 +08:00
3cd1109249
Add event-free survival evaluation
2026-06-27 11:48:47 +08:00
df6b45b95a
Clarify all-future model interpretation
2026-06-27 10:53:55 +08:00
6df0435f65
Refactor DeepHealth indices around disease expression
2026-06-26 16:25:36 +08:00
9ec3921bed
Add burden trajectory plotting script
2026-06-26 12:49:21 +08:00
fb873326ee
Keep burden landmark query times consistent
2026-06-26 12:24:51 +08:00
7ee29329f5
Write burden CSV shards in workers
2026-06-26 12:06:57 +08:00
263267f583
Use DataLoader for burden readouts
2026-06-26 11:50:12 +08:00
2b0b20d231
Use fork for Linux burden workers
2026-06-26 11:36:52 +08:00
c5ecbb79f2
Keep burden reduction on GPU
2026-06-26 11:27:44 +08:00
63df61e1dd
Batch burden index readout computation
2026-06-26 11:15:15 +08:00
ee7d363e3d
Support multi-GPU burden index computation
2026-06-26 10:51:56 +08:00
e400dab887
Add burden mapping files and landmark computation
2026-06-26 10:42:36 +08:00
a3bf3a0cd4
Add burden index computation interfaces
2026-06-25 17:15:56 +08:00
0138cbd95b
Add calendar-dated event index utilities
2026-06-22 17:36:04 +08:00
56421eb3f5
Add dist_mode parameter to evaluate_landmark_auc for improved flexibility in evaluation
2026-06-22 09:40:08 +08:00
615f395058
Refactor evaluation device handling in AUC scripts for improved clarity and flexibility
2026-06-20 12:13:58 +08:00
e5ecb714ba
Refactor loss computation and model input handling for improved clarity and efficiency
2026-06-20 11:26:03 +08:00
c3cac6dcea
Refactor dataset.py to remove caching functionality and related methods for improved simplicity
2026-06-18 16:25:55 +08:00
aa8ec5c3ac
Refactor code structure for improved readability and maintainability
2026-06-18 13:07:35 +08:00
1ea72e9133
Refactor DeepHealth model to expand extra-info token handling and update output structure
2026-06-17 14:27:07 +08:00
1757bcd25b
Refactor DeepHealth model and related components
...
- Removed BaselineEncoder and CrossAttention classes from models.py.
- Introduced OtherInfoTokenizer for handling additional token types.
- Updated DeepHealth class to integrate OtherInfoTokenizer and manage extra pooling logic.
- Added support for extra_pool_reduce parameter to control pooling behavior.
- Modified forward methods to return structured output using DeepHealthOutput dataclass.
- Updated training scripts to accommodate changes in model architecture and output handling.
- Enhanced error handling and validation for input shapes and types.
2026-06-17 11:05:10 +08:00
27aefb2f90
Refactor run directory creation to use a unique naming function in training scripts
2026-06-15 14:54:06 +08:00
36ec36c8a8
Add time attention mask handling and baseline class time computation to DeepHealth model
2026-06-15 14:35:10 +08:00
c3e49db859
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
593ecd2e71
Revert cross-attention integration into GPTBlock
2026-06-15 10:05:13 +08:00
c87c3b9f7c
Add cross-attention support to GPTBlock and update DeepHealth model integration
2026-06-13 17:02:04 +08:00
