Commit Graph

5 Commits

Author SHA1 Message Date
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
5e979e061b Add target construction and training script for DeepHealth model
- Implemented target construction in `targets.py` for next-token and unique-time set supervision.
- Added validation functions and utility methods for target building.
- Created a comprehensive training script in `train.py` that includes data loading, model building, optimizer setup, and training loop with early stopping and logging.
- Integrated loss functions and readout mechanisms based on target modes.
- Established dataset splitting and DataLoader configurations for training, validation, and testing.
2026-06-12 10:28:16 +08:00