Skip to content

Commit

Permalink
Refactor gsf.py to make the code of creating decoders and creating li…
Browse files Browse the repository at this point in the history
…nk prediction dataloders reusable. (#825)

*Issue #, if available:*
Related to #789 

*Description of changes:*
1. decouple the code of creating decoders from the code of creating
built-in models to make the code of creating decoders more reusable.
2. add `get_builtin_lp_train_dataloader_class` and
`get_builtin_lp_eval_dataloader_class` in gsf.py


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.

---------

Co-authored-by: Xiang Song <xiangsx@amazon.com>
  • Loading branch information
classicsong and Xiang Song authored May 6, 2024
1 parent 6054339 commit f0b949c
Show file tree
Hide file tree
Showing 4 changed files with 257 additions and 193 deletions.
2 changes: 2 additions & 0 deletions python/graphstorm/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,3 +28,5 @@
from .gsf import create_builtin_lp_model
from .gsf import create_builtin_edge_model
from .gsf import create_builtin_node_model
from .gsf import (get_builtin_lp_train_dataloader_class,
get_builtin_lp_eval_dataloader_class)
Loading

0 comments on commit f0b949c

Please # to comment.