Jiarui Li
46a3dfe628
- 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.
8 lines
109 B
Python
8 lines
109 B
Python
from __future__ import annotations
|
|
|
|
from evaluate_auc_v2 import main
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|