Skip to content

Commit

Permalink
Refactor gnn training (#253)
Browse files Browse the repository at this point in the history
* 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

* moved files

---------

Co-authored-by: anna-grim <anna.grim@alleninstitute.org>
  • Loading branch information
anna-grim and anna-grim authored Sep 27, 2024
1 parent 3c51a61 commit 51f39be
Show file tree
Hide file tree
Showing 8 changed files with 349 additions and 881 deletions.
330 changes: 0 additions & 330 deletions src/deep_neurographs/delete_merges_gt.py

This file was deleted.

Loading

0 comments on commit 51f39be

Please sign in to comment.