mirrors / dmlc--dgl

文件历史

410 提交 34 分支 42 Git 标签
0.2.x
提交图

3 次代码提交

作者 SHA1 备注 提交日期
Minjie Wang 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 f35ac5443e [Graph][Model] Cache adj & inc; MX GCN (#307) 
* add cache to adj and incmat

* Fix bug in cached adj/inc

* mx gcn spmv runnable; acc debugging...

* fix bug in mx gcn that loss is not correctly calculated

* fix mx utest

* fix as requested

* use raw parameter tensors rather than dense layer

* fix dropout

* Add numbers in readme
 2018-12-16 22:51:01 -05:00
Ziyue Huang cffa4034f5 [Model] fix GCN (#305) 
* mxnet gcn spmv

* update readme

* fix gcn

* pytorch gcn

* update readme
 2018-12-16 12:44:41 -05:00