Go to file

Jiarui Li a631ac6d59 feat: Add load_model function and update training script

Added a `load_model` function to `utils.py` to allow loading of trained models from configuration and state dictionary files.

The `train_iter.py` script was also modified, likely to incorporate or test this new functionality.

2025-10-18 11:07:59 +08:00

.gitattributes

feat: Add training and validation data via Git LFS

2025-10-16 14:24:56 +08:00

.gitignore

feat: Add multi-GPU training and improve config/ignore

2025-10-17 14:09:34 +08:00

GEMINI.md

feat: Implement time-aware GPT-2 for patient event prediction

2025-10-16 14:21:36 +08:00

LICENSE

Initial commit

2025-10-15 13:54:52 +08:00

models.py

feat: Add covariate-aware model and piecewise encoder

2025-10-17 12:04:50 +08:00

prepare_data.py

feat: Implement time-aware GPT-2 for patient event prediction

2025-10-16 14:21:36 +08:00

README.md

Initial commit

2025-10-15 13:54:52 +08:00

requirements.txt

refactor: Use AdamW optimizer and increase early stopping patience

2025-10-17 10:31:12 +08:00

train_iter_multigpu.py

feat: Add iteration-based training scripts (single and multi-GPU)

2025-10-18 10:05:37 +08:00

train_iter.py

feat: Add load_model function and update training script

2025-10-18 11:07:59 +08:00

train_multigpu.py

config: Tune hyperparameters for multi-GPU training

2025-10-17 15:37:42 +08:00

train.py

feat: Add command-line arguments to train.py

2025-10-18 10:23:12 +08:00

ukb_real_train.bin

feat: Add training and validation data via Git LFS

2025-10-16 14:24:56 +08:00

ukb_real_val.bin

feat: Add training and validation data via Git LFS

2025-10-16 14:24:56 +08:00

utils.py

feat: Add load_model function and update training script

2025-10-18 11:07:59 +08:00

README.md

DeepHealth