02eb463ab1
report accuracy and fix training mask ( #181 )
2018-11-27 15:15:57 +08: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
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
9e9a9488f1
[GraphIndex] Immutable graph index using MXNet CSRArray ( #86 )
...
* add immutable graph index.
* update ImmutableGraphIndex.
* add benchmarks of subgraph generation.
* fix node_subgraphs.
* use scipy synthetic data in immutable graph.
* create immutable_graph.
* use spmv in gcn.
* fix immutable graph index.
* update graph index.
* create ImmutableSubgraphIndex.
* return subgraph node and edge mapping.
* fix benchmark.
* add mapping to subgraph nid.
* test in_edges and out_edges.
* update benchmark.
* Fix immutable subgraph.
* complete in_edges and out_edges.
* fix benchmarks.
* add test on node_subgraphs
* add sse.
* construct subgraphs in parallel.
* add in/out degree(s).
* make immutable graph index backend-specific.
* implement has_edge(s) and edge_id(s).
* Revert "use spmv in gcn."
This reverts commit 9cfed5f5fb3dd2ed9b98745348b1c0e9731ed7f7.
* implement node_subgraphs in GraphIndex.
* fix sse.
* address comments.
* address comments.
* Update mxnet docker.
2018-10-26 11:28:25 -07:00
5567f4a4d9
[BACKEND] Add MXNet backend. ( #77 )
...
* support mxnet.
* add mxnet version of GCN.
* rename mxnet.nd as F.
* add mxnet GAT.
* enable GPU for GCN.
* fix MXNet GCN train.
* Use adam to optimize GAT
* support more operators.
* support sparse arrays.
* update mxnet backend.
* support index_copy.
* remove NN.
* update mxnet backend.
* temp check in.
* fix data conversion.
* add test.
* clean up mxnet backend.
* update mxnet examples.
* Revert "remove NN."
This reverts commit d815d9a0ec619f9ce9099c48cd35db9d8e947483.
* temp disable MXNet version of NN.
2018-10-14 11:15:48 -04:00
15a2c22c98
Reorg folders; basic impl of the APIs
2018-06-14 21:37:43 -04:00
yifeim