Skip to content

Inference and Training Pipelines

Latest
Compare
Choose a tag to compare
@anna-grim anna-grim released this 27 Sep 20:21
· 79 commits to main since this release
3c51a61
Refactor gnn training (#252)

* minor upds

* refactor: training pipeline

* feat: find gcs image path

* feat: feature generation in trainer

* feat: validation sets in training

* bug: hgraph forward passes with missing edge types

* refactor: hgnn trainer

* feat: functional training pipeline

* bug: set validation data

* refactor: combined train engine and pipeline

* refactor: infernce pipeline, evaluation

---------

Co-authored-by: anna-grim <anna.grim@alleninstitute.org>