Skip to content

GitLab

Unverified Commit a1a492ae authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Merge pull request #3 from SherylHYX/master 

init ScalableGNN
parents b4dec6bb ffc1ca90
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
torch_geometric_autoscale/__init__.py
View file @ a1a492ae
......@@ -15,6 +15,7 @@ from .pool import AsyncIOPool # noqa
from .metis import metis, permute # noqa
from .utils import compute_micro_f1, gen_masks, dropout # noqa
from .loader import SubgraphLoader, EvalSubgraphLoader # noqa
from .models import ScalableGNN
__all__ = [
'get_data',
......@@ -27,5 +28,6 @@ __all__ = [
'dropout',
'SubgraphLoader',
'EvalSubgraphLoader',
'ScalableGNN',
'__version__',
]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment