063ed69cdf
[BUGFIX] fix a bug in creating immutable graph index. ( #251 )
...
* fix a bug in creating immutable graph index.
* fix for new changes in the backend API.
* fix for creating immutable graph index from coo matrix.
* retrigger
2018-12-05 17:51:50 -05:00
899d125bb1
Add tests for MXNet and throw NotImplementedError if not implemented ( #258 )
...
* add more unit tests for mxnet.
* fix.
2018-12-05 10:41:33 -08:00
eafcb7e7f5
[Bugfix][MXNet] Fix edge order and builtin max bug in mx ( #247 )
...
* Fix edge order and builtin max bug in mx
* fix as requested
2018-12-05 01:32:54 -08:00
c99f423718
[Sampler] extend sampler ( #238 )
2018-12-04 16:46:26 -05:00
a5a35d1f7a
[Sampler] improve random shuffle performance in sampler ( #228 )
...
* fix.
* make it generic.
* add the API.
* fix.
* remove mxnet.
2018-12-04 09:24:32 -08:00
57b07fce92
[Model] Accelerate SSE ( #172 )
...
* lazy eval edge ids.
* parallelize node id lookup.
* fix a bug.
* use mxnet for index.
* use update_all in the subgraph training.
* sample neighbors.
* Revert "parallelize node id lookup."
This reverts commit e661f69bb06cb5a7c246f0e106f245e27800e220.
* update README.
* cache subgraphs.
* support all degrees.
* cache adj in CPU.
* fix a bug in sse.
* print.
* raise error on mutable graphs.
* measure train time per epoch.
* fix a bug in graph_index.
* remove readonly in DGLSubGraph.
* cache subgraph properly.
* accelerate getting adjacency.
* split infer.
* lazy eval edges.
* specify inference mode.
* update for new sampler.
* use new mxnet sampling api.
* fix indent.
* remove profiling code.
* remove mxnet from sampler.
* return a lambda function for graph edges.
* add docs for immutable subgraph.
* Revert "return a lambda function for graph edges."
This reverts commit 0de5d7f100e230c518a3fb8976a6227f474d09ee.
* get parent_eid.
2018-12-02 12:43:25 -05:00
9e8261735f
[Doc] Minor doc update for mxnet/immutable_graph_index ( #194 )
2018-11-29 20:59:13 -05:00
5f89cd662e
Fix the convergence problem in SSE. ( #162 )
...
* add a test.
* move file.
* debug
* add degree normalization.
* it can converge now.
* remove hidden_data.
* use readonly graph index.
* use subgraph loader.
* refactor code and remove intermediate data.
* split inference.
* mix cpu and gpu training.
* fix mxnet sparse_matrix constructor.
* convert tensor context.
* set up the prediction model.
* load mxnet csr directly.
* add timing.
* move test_sse.py to sse_batch.py
* fix the tensor version of SSE.
* update README.
2018-11-20 15:57:52 +08:00
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
048f6d7a30
[GraphIndex] refactor graph caching ( #150 )
...
* refactor graph caching
* fix mx test
* fix typo
2018-11-13 22:12:38 -05:00
40ca5de450
[API] Readout interfaces ( #124 )
...
* fixing builtin src*edge shape mismatch
* bundled function refactor (?)
* fixing names
* readout prototype
* oops
* more fixes
* removing readout prototype
* sum_on() with SPMV, fixing batching with 0 edges
* readouts with segmented sum
* typo (??????)
* fixes NLTK dependency (#125 )
* misc fixes including #126 (pushing again)
* sanity check for mxnet
* fixes NLTK dependency (#125 ) and #126
* reverting to sum_nodes/edges
2018-11-10 11:35:34 -05:00
bd0e4fa0b3
[MXNet][API] move to the new API ( #123 )
...
* move gat to the new api.
* fix gcn.
* update sse.
* fix dgl core.
* update sse.
* fix small bugs in dgl core.
* fix mxnet tests.
* retrigger
* address comments and fix more bugs.
* fix
* fix tests.
2018-11-08 10:00:47 -08:00
7241a9c008
[Backend] backend interface ( #109 )
...
* backend interface
* small fix
* more comments to the data type dict
* WIP
* convert_to and narrow
* WIP
* pytorch and numpy backend; WIP on mxnet backend
* mxnet backend
* narrow
* Fix all usages
* fix for mx
* fix for mx
* fix mx
* fix mx
* fix mx
* fix mx
* fix mx
* fix mx
* fix mx
* revert jenkins
* add sparse_matrix api
* sparse matrix api
* some fixme
* Fix as requested
2018-11-04 21:42:39 -05:00
Da Zheng