dmlc--dgl
a7e941c379
* Add sparse embedding for dgl and update rgcn example * upd * Fix * Revert "Fix" This reverts commit 4da87cdfb8b8c3506b7fc7376cd2385ba8045c2a. * Fix * upd * upd * Fix * Add unitest and update impl * fix * Clean up rgcn example code * upd * upd * update * Fix * update score * sparse for sage * remove model sparse * upd * upd * remove global norm * revert delete model_sparse.py * update according to comments * Fix doc * upd * Fix test * upd * lint * lint * lint * upd * upd * clean up Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
9 行
236 B
Python
9 行
236 B
Python
"""Package for pytorch-specific NN modules."""
|
|
from .conv import *
|
|
from .glob import *
|
|
from .softmax import *
|
|
from .factory import *
|
|
from .hetero import *
|
|
from .utils import Sequential, WeightBasis
|
|
from .sparse_emb import NodeEmbedding
|