 Minjie Wang
|
708765f0a1
|
[NN] RGCN modules (#744)
* rgcn module
* support id input
* WIP: model codes
* use faster index select
* dropout
* self loop
* WIP: link prediction
* fix lint
* WIP: docs
* docstring
* docstring
* merge two child classes
* mxnet rgcn module
* fix lint
* fix lint
* fix rename bug
* add uniform edge sampler
* fix fn name
* docstring
* fix mxnet rgcn module
* fix mx rgcn
* enable test on cuda
|
2019-08-23 16:38:48 -04:00 |
|