Zihao Ye
af61e2fbb4
[Feature] Support nn modules for bipartite graphs. ( #1392 )
...
* init gat
* fix
* gin
* 7 nn modules
* rename & lint
* upd
* upd
* fix lint
* upd test
* upd
* lint
* shape check
* upd
* lint
* address comments
* update tensorflow
Co-authored-by: Quan Gan <coin2028@hotmail.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2020-03-29 16:48:38 +08:00
Jinjing Zhou
978ac41097
[NN] Fix RelGraphConv num_bases error message ( #1309 )
...
* Fix RelGraphConv when num_bases is None
* Improve error message in when num_bases is wrong
* Fix integer modulo by zero exception when num_bases is 0
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2020-03-06 01:00:48 +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