mirrors / dmlc--dgl

文件历史

919 提交 34 分支 42 Git 标签
0.4.x
提交图

3 次代码提交

作者 SHA1 备注 提交日期
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
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
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