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
1c4bfb62bb
[Performance] Replace np.array with np.asarray ( #1301 )
...
* replace np.array with np.asarray
* fix
Co-authored-by: Minjie Wang <minjie.wang@nyu.edu >
2020-03-03 20:45:41 +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