文件历史

提交图

634 次代码提交

作者 SHA1 备注 提交日期
GaiYu0 52ed6a4507 train.py help 2018-10-06 16:27:16 +08:00
GaiYu0 cc372e3774 sbm mixture 2018-10-04 21:22:41 +08:00
Minjie Wang 8e71b64e2a gat runnable 2018-10-03 21:17:23 -04:00
Minjie Wang fde4f581d2 gcn runnable 2018-10-03 21:05:19 -04:00
GaiYu0 f52fc3fcdf experiment 6.1 2018-10-01 09:12:50 +08:00
GaiYu0 2b09281178 line graph 2018-10-01 09:12:35 +08: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
Lingfan Yu c42eac718f Fix batched graph edge order bug and other fixes (#50)
* fix dgl.batch edge ordering bug

* add graph batching test cases

* fix partial spmv ctx.

* add dataset generating for dgmg
2018-08-23 15:11:38 -04:00
Minjie Wang 6105e44142 Many fix and updates (#47)
* subgraph copy from

* WIP

* cached members

* Change all usage of id tensor to the new Index object; remove set device in DGLGraph;

* subgraph merge API tested

* add dict type reduced msg test
2018-08-22 12:25:05 -04:00
Lingfan Yu 1f16f29b20 bug fix 2018-08-16 19:31:43 +00:00
Lingfan Yu 96179b0c96 Deep Generative Models of Graphs (#14)
* model code for generative graphs

* batched version for dynamic graph generation using padding

* renaming function train back to forward

* remove old util function for padding DGMG

* override networkx clear to reset state, add dgl.nn

* Dynamic graph without batching

* use relative import path

* load dataset, pad batch

* bug fix

* experimental batch and unbatch

* dgmg batched version

* minor tweak

* move preprocessing padding into data loading

* batch graph test code

* minor

* batched graph class and test cases

* make dgl.nn.gcn a simple layer plus minor fix

* update dgmg model

* test forward using attribute field

* use frame append, minor changes

* moving networkx operations out of forward

* revert some changes

* remove structural immutability check
2018-08-16 14:05:50 -04:00
Minjie Wang ee241699e0 GAT model (#37)
* GAT model

* fix output projection to have only one head
2018-08-09 22:44:53 -04:00
Minjie Wang 0a78dbe12a Three DGL-GCN implementations (#35)
* synthetic dataset

* some fix

* gcn readme
2018-08-07 15:26:58 -04:00
Minjie Wang 2c489fadec SPMV specialization (#32)
* fix edge list order problem in cached graph.

* minor fix

* fix bug in edge iter

* SPMV works

* gcn spmv on CPU

* change gcn style

* fix cached graph performance; fixed gcn dataset bug

* reorg dir

* non-batch spmv; partial update problem with shape change

* fix reorder problem; finish gcn-batch impl

* pop API

* GPU context
2018-08-06 14:00:06 -04:00
Ivan Brugere 57f480f591 LSTM fix (#28)
Correctly handling lstm model creation
2018-07-13 01:05:59 -04:00
Minjie Wang c50b90cf7e Change recvfrom API; fix corner cases in issue20 (#25)
* Change ; fix corner cases in issue20

* Change recvfrom to recv
2018-07-12 11:22:38 -04:00
Ivan Brugere 0d6504434b cleaning
cleaning and documentation
2018-07-10 15:57:57 -07:00
Ivan Brugere ef18dab2a1 initial ginipath
Initial example file for ginipath
2018-07-10 12:35:13 -07:00
Lingfan Yu 68fb5f7e51 Reduce API (#15)
* add reduce_msg related api to dgl graph

* add reduce_sum, switch backend from numpy to pytorch

* update gat gcn to use reduce msg api

* remove reduce_sum

* add built-in reduce functions
2018-07-04 11:09:08 +09:00
GaiYu0 b9073209f1 CDGNN example & minor fixes in dgl/graph.py (#11)
* cdgnn example & minor fixes in dgl/graph.py

* incomplete TreeLSTM

* add data preprocess of SST

* fix nx_SST.py

* TreeLSTM completed
2018-06-25 14:55:42 +09:00
Lingfan Yu c87564d553 minor bug fixes 2018-06-22 20:13:44 +00:00
Lingfan Yu 20d5d5321b readme for gcn and gat dataset 2018-06-20 04:01:22 +00:00
Lingfan Yu 9f0d9e4246 example gat model 2018-06-20 04:00:53 +00:00
Lingfan Yu 1248bd24bb example gcn model 2018-06-20 04:00:44 +00:00
Lingfan Yu 34fac23d2f rename gan.py to gat.py 2018-06-20 04:00:18 +00:00
zzhang-cn 174c1d5516 add edge repr func; modify mgcn.py 2018-06-19 13:10:09 +08:00
Minjie Wang fb214b3059 Add mgcn example for discussion 2018-06-18 10:48:14 -04:00
Minjie Wang 2455934ac4 checkin runnable topdown model in the new API 2018-06-15 20:47:30 -04:00
Minjie Wang 5b5baee165 add topdown example 2018-06-15 05:13:23 -04:00
Minjie Wang 15a2c22c98 Reorg folders; basic impl of the APIs 2018-06-14 21:37:43 -04:00