Commit Graph

64 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
58f253d9b6 Implement parallel processing for DOA AUC evaluation and enhance task management 2026-06-13 15:45:07 +08:00
76787d2fb2 Add evaluation split handling and dataset subset size to DOA AUC evaluation 2026-06-13 15:39:31 +08:00
46a3dfe628 Add training scripts for all-future and next-step supervision with DeepHealth
- Implement `train_all_future.py` for training with query-conditioned all-future supervision.
- Implement `train_next_step.py` for training with next-token/next-time-point supervision.
- Introduce `train_util.py` for shared utility functions including logging, dataset splitting, and model checkpointing.
- Enhance argument parsing for both training scripts to accommodate new parameters.
- Update loss functions and model configurations to support the new training paradigms.
2026-06-13 11:42:04 +08:00
034d8065a7 Refactor bias handling in AUC evaluation scripts to improve robustness and prevent errors when bias is not defined 2026-06-12 17:28:22 +08:00
1240766aa9 Enhance training script and evaluation logic to support all-future model target mode and improve error handling for distribution modes 2026-06-12 11:53:35 +08:00
82f70945d9 Enhance training script to support all-future model target mode and update dataset handling 2026-06-12 11:49:44 +08:00
3125b6119f Refactor AUC evaluation scripts to support model target modes and improve distribution handling 2026-06-12 11:33:32 +08:00
0fa8bbbb9a Enhance data parsing and validation, add extra info types files
- Improved `parse_int_list` and `parse_float_list` functions to support JSON list input.
- Introduced `validate_dataset_metadata` function to ensure dataset metadata consistency with training configuration.
- Added multiple new files for extra information types, categorizing them into assessment-only, exposure-only, and combined types.
- Removed deprecated `merge_extra_info_types` function and adjusted related logic in `train.py`.
- Updated `save_config` function to accept additional metadata for training runs.
- Refactored model and training scripts for better clarity and maintainability.
2026-06-12 11:16:19 +08:00
fc8c7b7177 Refactor code structure for improved readability and maintainability 2026-06-12 10:40:48 +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