gemeo-twin-stack / src /gemeo /train /__init__.py
timmers's picture
GEMEO world-model β€” initial release (module + NeuralSurv ckpt + RareBench v49 + KG embeddings)
089d665 verified
raw
history blame contribute delete
631 Bytes
"""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`.
"""