Go to file

Jiarui Li 4181ead03a Refactor: Improve attention mechanism and early stopping

- Refactor the self-attention mechanism in `models.py` to use `nn.MultiheadAttention` for better performance and clarity.
- Disable early stopping check during warmup epochs in `train.py` to improve training stability.

2025-10-16 15:57:27 +08:00

.gitattributes

feat: Add training and validation data via Git LFS

2025-10-16 14:24:56 +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

Refactor: Improve attention mechanism and early stopping

2025-10-16 15:57:27 +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

train.py

Refactor: Improve attention mechanism and early stopping

2025-10-16 15:57:27 +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: Implement time-aware GPT-2 for patient event prediction

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

README.md

DeepHealth