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
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
5d3f470b72
[Feature] DGL Pooling modules ( #669 )
...
* removal doc
* glob
* upd
* rm knn
* add softmax
* upd
* upd
* add broadcast and s2s
* optimize max_on
* forsaken changes to heterograph
* upd
* upd
* upd
* upd
* upd
* bugfix
* upd
* upd
* upd
* upd
* format upd
* upd format
* upd doc
* upd
* import order
* upd
* rm warnings
* fix
* upd test
* upd
* upd
* fix device
* upd
* upd
* upd
* upd
* remove 1.1
* upd
* trigger
* trigger
* add more tests
* fix device
* upd
* upd
* refactor
* fix?
* fix
* upd docstring
* refactor
* upd
* fix
* upd
* upd
* upd
* fix
* upd docs
* add shape
* refactor & upd doc
* upd doc
* upd
2019-08-06 01:43:25 +08:00
74e13eea61
[Model] Update GAT model code ( #622 )
...
* fix gat code to use latest edge softmax module
* avoid transpose
* update README
* use edge_softmax op
* mxnet edge softmax op
* mxnet gat
* update README
* fix unittest
* fix ci
* fix mxnet nn test; relax criteria for prod reducer
2019-06-08 21:41:38 -04:00
565f0c88fc
[WIP] [NN] Refactor NN package ( #406 )
...
* refactor graph conv
* docs & tests
* fix lint
* fix lint
* fix lint
* fix lint script
* fix lint
* Update
* Style fix
* Fix style
* Fix style
* Fix gpu case
* Fix for gpu case
* Hotfix edgesoftmax docs
* Handle repeated features
* Add docstring
* Set default arguments
* Remove dropout from nn.conv
* Fix
* add util fn for renaming
* revert gcn_spmv.py
* mx folder
* fix wierd bug
* fix mx
* fix lint
2019-02-25 18:41:21 -05:00
Minjie Wang