文件历史

提交图

1492 次代码提交

作者 SHA1 备注 提交日期
Minjie Wang 314a75f367 WIP: batched graph 2018-09-24 17:31:50 -04:00
Minjie Wang 79f62400fe WIP: batch graphs 2018-09-21 14:53:25 -04:00
Minjie Wang 4aebfd7bd8 add merge graph; fix bug in from_networkx 2018-09-21 12:25:54 -04:00
Minjie Wang 7b5d4c5886 pass specialization test 2018-09-20 18:22:46 -04:00
Minjie Wang b2e4bdc03f passed basic test batching 2018-09-20 14:44:06 -04:00
Gan Quan f26ecb4909 Fix batching node-only graphs (#62)
* fixing batching with graphs with no edges

* oops forgot test
2018-09-20 10:36:35 -04:00
Minjie Wang 44db98c4e0 remove nx inheritence 2018-09-19 22:49:14 -04:00
Minjie Wang 916d375ba3 Merge branch 'master' into cpp 2018-09-19 21:04:31 -04:00
Minjie Wang a1038eb1ee graph index 2018-09-19 21:02:15 -04:00
Lingfan Yu 9b0a01db41 Support list of msg or reduce func for update_all and send_and_recv (#58)
* support multiple fields in spmv

* finishing SPMV executor

* non-spmv case

* refactor code to give single mfunc rfunc a shortcut

* two test cases to test multiple msg/red func

* catching cases where msg uses anonymous field

* default ALL for update edge

* more corner case test

* print failed test

* delete print

* fix builtin max reducer
2018-09-19 20:49:29 -04:00
Minjie Wang a81d27dc67 WIP: integrating 2018-09-19 10:52:20 -04:00
Minjie Wang dba79f1d45 also return edge id for many APIs 2018-09-18 23:40:13 -04:00
Minjie Wang 71e6e0cd5f python shim + some simple tests 2018-09-17 14:40:29 -04:00
Minjie Wang 758cb16e67 ndarray argument 2018-09-17 00:15:46 -04:00
Minjie Wang c086d454d3 Use TVMContext 2018-09-16 16:56:29 -04:00
Minjie Wang b24daa6607 change to rel import within dgl 2018-09-16 16:14:37 -04:00
Minjie Wang 842d37682b python api 2018-09-16 15:49:11 -04:00
Minjie Wang 2694b12725 import ffi solution from TVM 2018-09-05 10:51:31 -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
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
Gan Quan 5e75f5dbfc Fix reducing with 0 incoming messages (#48)
* fixing 0 degree reduces

* adopting accum=None semantics

* minor fix as per reviews

* TODO in test

* fix test
2018-08-24 11:51:05 -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
Minjie Wang 3721822e9e Subgraph API (#39)
* subgraph

* more test cases

* WIP

* new FrameRef and test

* separate nx init code

* WIP

* subgraph code and test

* line graph code and test

* adding new test for adding new features on line graphs

* no backtracking line graph

* fix inplace relabel
2018-08-18 14:14:43 -04: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 e3bac70b4f Spmv partial (#43)
* partial spmv impl and test

* some fix for update edge
2018-08-12 21:02:55 -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 4673b96f92 fix set_n_repr grad problem (#38) 2018-08-09 17:48:55 -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
Minjie Wang 11e42d1094 Batching semantics and naive frame storage (#31)
* batch message_func, reduce_func and update_func

Conflicts:
	python/dgl/backend/pytorch.py

* test cases for batching

Conflicts:
	python/dgl/graph.py

* resolve conflicts

* setter/getter

Conflicts:
	python/dgl/graph.py

* test setter/getter

Conflicts:
	python/dgl/graph.py

* merge DGLGraph and DGLBGraph

Conflicts:
	python/dgl/graph.py

Conflicts:
	python/dgl/graph.py

* batchability test

Conflicts:
	python/dgl/graph.py

Conflicts:
	python/dgl/graph.py

* New interface (draft)

Conflicts:
	_reference/gat_mx.py
	_reference/molecular-gcn.py
	_reference/molecular-gcn_mx.py
	_reference/multi-gcn.py
	_reference/multi-gcn_mx.py
	_reference/mx.py
	python/dgl/graph.py

* Batch operations on graph

Conflicts:
	.gitignore
	python/dgl/backend/__init__.py
	python/dgl/backend/numpy.py
	python/dgl/graph.py

* sendto

* storage

* NodeDict

* DGLFrame/DGLArray

* scaffold code for graph.py

* clean up files; initial frame code

* basic frame tests using pytorch

* frame autograd test passed

* fix non-batched tests

* initial code for cached graph; tested

* batch sendto

* batch recv

* update routines

* update all

* anonymous repr batching

* specialize test

* igraph dep

* fix

* fix

* fix

* fix

* fix

* clean some files

* batch setter and getter

* fix utests
2018-07-30 14:10:37 -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
Minjie Wang d24ccfdda6 Anonymous node/edge repr (#24)
* Remove name args in register/sendto/update_edge/recvfrom

* Support anonymous repr

* Test anonymous edge repr

* Test node states with both anonymous and explicit reprs
2018-07-11 15:15:47 -04:00
Minjie Wang 9219349a8f Use edge update to impl sendto; fix examples with missing reduce func (#22) 2018-07-10 13:18:24 +09: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
zzhang-cn 174c1d5516 add edge repr func; modify mgcn.py 2018-06-19 13:10:09 +08:00
Minjie Wang 8e2e68df2b fix bugs for topdown.py models 2018-06-15 20:44:25 -04:00
Minjie Wang 6858ed54fd WIP: topdown example 2018-06-15 12:54:16 -04:00
Minjie Wang aa5bd89f09 minor change on the supported container type 2018-06-15 05:08:38 -04:00
Minjie Wang f310e58613 Add test/example; fix bug in graph.py 2018-06-14 23:21:26 -04:00
Minjie Wang 15a2c22c98 Reorg folders; basic impl of the APIs 2018-06-14 21:37:43 -04:00