Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[NN] Refactor the Code Structure of GT #5100

Merged
merged 15 commits into from
Apr 14, 2023
Merged

[NN] Refactor the Code Structure of GT #5100

Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
6a975e1
refactor code structure
rudongyu Jan 4, 2023
20a02ea
update
rudongyu Jan 4, 2023
a714bbc
update
rudongyu Jan 4, 2023
963cae3
lint
rudongyu Jan 4, 2023
f8243b0
fix C0103
rudongyu Jan 30, 2023
5c3584a
lint
rudongyu Mar 14, 2023
5247a46
fix attention pollution issue
rudongyu Mar 29, 2023
ef59998
Merge branch 'master' into gt_refactor
rudongyu Mar 30, 2023
582ae71
fix
rudongyu Mar 31, 2023
a1d5f71
Merge branch 'gt_refactor' of github.com:rudongyu/dgl into gt_refactor
rudongyu Mar 31, 2023
2e9900c
update
rudongyu Apr 12, 2023
6acb034
update
rudongyu Apr 12, 2023
7b80a95
update
rudongyu Apr 12, 2023
edf420c
lint
rudongyu Apr 13, 2023
7197357
Merge branch 'master' into gt_refactor
rudongyu Apr 13, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 15 additions & 0 deletions docs/source/api/python/nn-pytorch.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -145,3 +145,18 @@ Network Embedding Modules

~dgl.nn.pytorch.DeepWalk
~dgl.nn.pytorch.MetaPath2Vec

Utility Modules for Graph Transformer
----------------------------------------
.. autosummary::
:toctree: ../../generated/
:nosignatures:
:template: classtemplate.rst

mufeili marked this conversation as resolved.
Show resolved Hide resolved
~dgl.nn.pytorch.gt.DegreeEncoder
~dgl.nn.pytorch.gt.LapPosEncoder
~dgl.nn.pytorch.gt.PathEncoder
~dgl.nn.pytorch.gt.SpatialEncoder
~dgl.nn.pytorch.gt.SpatialEncoder3d
~dgl.nn.pytorch.gt.BiasedMHA
~dgl.nn.pytorch.gt.GraphormerLayer
10 changes: 2 additions & 8 deletions python/dgl/nn/pytorch/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,12 +8,6 @@
from .factory import *
from .hetero import *
from .sparse_emb import NodeEmbedding
from .utils import (
JumpingKnowledge,
LabelPropagation,
LaplacianPosEnc,
Sequential,
WeightBasis,
)
from .utils import JumpingKnowledge, LabelPropagation, Sequential, WeightBasis
from .network_emb import *
from .graph_transformer import *
from .gt import *
Loading