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
6111ea4665
[Model] RGCN mini-batch training; Better bipartite graph support ( #1337 )
...
* change the model to use node embedding only
* minibatch training
* add readme
* small fix
* backward-compatible
* backward compatible
* modify to_block; rgcn changed
* fix
* fix transform
* fix bug in unittest script
* docstring
* fix lint
* add tests
* address comments; fix offline eval
* gitignore
2020-03-16 18:58:06 +08:00
7a80faf120
[Bugfix] Correct the loss function in RGCN model ( #1217 )
...
* upd
* upd
* fix
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2020-02-13 18:35:07 +08:00
77ae4d5321
[Example] Fix Rgcn hetero nonedetermistic in graph and model initialization ( #1244 )
...
* Fix model save and load problem.
Add offline inference example
* Fix test
* Fix randomness problem between test runs
* Make rdf loader deterministic
2020-02-07 11:07:09 +08:00
1c91f460d3
[Hetero][Model] RGCN for heterogeneous input ( #885 )
...
* new hetero RGCN
* bgs running
* fix gpu
* am dataset
* fix bug in label preparation
* Fix AM training; add result
* rm sym link
* new embed layer; mutag
* mutag matched; other fix
* minor fix
* dataset refactor
* new data loading
* rm old files
* refactor
* docstring
* include literal nodes in AIFB dataset
* address comments
* docstring
2019-09-29 00:21:17 -04:00
Minjie Wang