 Minjie Wang
|
3efb5d8ecf
|
[NN] Add HeteroGraphConv module for cleaner module definition (#1385)
* Add HeteroGraphConv
* add custom aggregator; some docstring
* debugging
* rm print
* fix some acc bugs
* fix initialization problem in weight basis
* passed tests
* lint
* fix graphconv flag; add error message
* add mxnet heteroconv
* more fix for mx
* lint
* fix torch cuda test
* fix mx test_nn
* add exhaust test for graphconv
* add tf heteroconv
* fix comment
|
2020-03-27 18:47:06 +08:00 |
|
|
Zihao Ye
|
aa06e4579f
|
[Doc] Improve the docstring of several nn modules. (#1187)
* upd
* upd
* upd
* upd
* lint
* upd
* upd
* upd
* upd
* upd
Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
|
2020-01-09 21:14:33 +08:00 |
|
|
Zihao Ye
|
ca302a13ac
|
[Feature] Add dgl.nn.*.Sequential for usability (#1166)
* upd
* upd
* upd
* upd
* lint
* upd
* upd
* upd
* upd
Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
|
2020-01-09 12:39:22 +08:00 |
|
|
VoVAllen
|
ddb5d804f9
|
[Refactor] Break NN modules into files (#859)
* break nn modules into files
* break mxnet nn modules
* fix lint
* fix lint
|
2019-09-16 14:44:15 +08:00 |
|
|
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 |
|