2389df8155
[API] Subgraph sampling API. ( #138 )
...
* add neighbor sampler.
* fix sampler.
* handle non-existing nodes differently in MapParentIdToSubgraphId.
* fix sampling tests.
* parallelize sampling.
* add more tests.
* fix a bug in subgraph loader.
* get number of sampled nodes directly.
* load mxnet CSR to DGLGraph directly.
* create graph from mxnet csr array directly.
* use subgraph loader in SSE.
* fix a bug in sampling.
* output framework tensor.
* fix.
* remove unnecessary code.
* fix tests.
* add SSE README.
* move to contrib.
* address comments.
* add tests.
2018-11-14 13:51:27 -08:00
cc372e3774
sbm mixture
2018-10-04 21:22:41 +08:00
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
0a78dbe12a
Three DGL-GCN implementations ( #35 )
...
* synthetic dataset
* some fix
* gcn readme
2018-08-07 15:26:58 -04:00
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
Da Zheng