File size: 631 Bytes
089d665 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | """Phase 2 training pipelines.
These modules require torch_geometric and a GPU. They produce checkpoints
under `gemeo/artifacts/` that the runtime modules pick up automatically.
Run order:
1. python -m gemeo.train.primekg — ingest PrimeKG into our schema
2. python -m gemeo.train.hgt — train patient HGT encoder
3. python -m gemeo.train.txgnn — fine-tune TxGNN for drug repurposing
4. python -m gemeo.train.tgnn — train trajectory TGN over snapshot chains
5. python -m gemeo.train.neuralsurv — train survival head
All scripts can be re-run independently. See `gemeo/train/README.md`.
"""
|