文件历史

14 次代码提交

作者 SHA1 备注 提交日期
Minjie Wang 3564fdc5f1 [Bugfix][Model] fix treelstm model (#274)
* fix bug after moving batcher out of dgl.data

* disable mx utest
2018-12-06 23:52:50 -05:00
Da Zheng 1bbc885b40 move batcher to examples. (#269)
* move pytorch code to examples.

* fix.

* fix tutorial
2018-12-06 06:45:27 -08:00
Minjie Wang 21255b6528 [Bugfix] tolist and dependencies in dgl.data (#239)
* change Index.tolist -> Index.tonumpy; fix bug in traversal; remove dependencies in data

* fix import

* fix __all__ and some docstring
2018-12-05 12:20:59 -05:00
Zihao Ye c80dc5e012 [model] Tree-LSTM update & hotfix(dropbox link) (#202)
* change the signature of node/edge filter

* upd filter

* Support multi-dimension node feature in SPMV

* stable version

* hotfix

* upd tutorial

* upd README
2018-11-30 19:39:49 -05:00
Zihao Ye 53b9a4bdbc [DOC][Model] Update Tree-LSTM (#152)
* update tree lstm

* tree_lstm (new interface)

* simplify pop

* merge qipeng(root)

* upd tree-lstm & tutorial

* upd model

* new capsule tutorial

* capsule for new API

* fix deprecated API

* New tutorial and example

* investigate gc problem

* add viz code

* new capsule tutorial

* remove ipynb

* move u_hat

* add link

* add requirements.txt

* remove ani.save

* update ci to install requirements

* utf-8

* change seed

* graphviz requirement

* accelerate

* little format

* update some markup
2018-11-14 22:29:59 -05:00
Zihao Ye a9ffb59e66 [DOC][Model] Tree-LSTM example and tutorial (#146)
[DOC][Model] Tree-LSTM example and tutorial
2018-11-13 18:09:14 -05:00
Minjie Wang 8ea359d125 [API] Message propagation APIs (#127)
* add examples in traversal.py

* message propagate methods

* use the new message propagation for tree-lstm

* update to the new name

* update propagate API doc

* update doc

* add propagate utest
2018-11-09 15:04:57 -05:00
Zihao Ye 1eb17bb01f [Model] Adapt Tree-LSTM to new interface (#122)
* tree_lstm (new interface)

* simplify pop
2018-11-08 10:31:55 -05:00
Minjie Wang 1f0f57b300 fix bug in treelstm 2018-10-08 17:29:28 +00:00
Minjie Wang d772d390b2 fix 2018-09-25 03:55:18 +00:00
Minjie Wang a8a4fcbaff quickly integrating with tree-lstm example 2018-09-24 23:42:56 -04:00
Minjie Wang 61fa3c6cf5 Builtin function and API changes (#53)
* WIP: API renaming

* API rewrite and node function refactor

* builtin functions

* builtin functions tested

* fix test

* send and recv spmv test

* WIP: fix examples

* Fix examples using new APIs
2018-09-01 13:38:01 -04:00
Minjie Wang 8c71f3f86d Fix treelstm CPU 2018-08-29 10:53:22 -04:00
Gan Quan b7eb16595f Fix 0deg update_to and Tree-LSTM model (#51)
* WIP

* WIP

* treelstm dataloader

* Main training loop.

* trainable treelstm script

* fix dependency

* cuda training

* Add tensorized topological traversal

* allowing update_to() with no incoming messages

* fixing partial cases
2018-08-28 11:57:08 -04:00