1. 09 Jun, 2019 1 commit
    • Minjie Wang's avatar
      [Examples] Run all the examples (#623) · dec8b49b
      Minjie Wang authored
      * all pytorch examples
      
      * scan through mxnet examples
      
      * change reddit data
      
      * tweak numerical range for unittest
      
      * fix ci
      
      * fix ci
      
      * fix
      
      * add seed to workaround
      dec8b49b
  2. 28 Feb, 2019 1 commit
  3. 19 Feb, 2019 1 commit
    • Da Zheng's avatar
      [Feature] add NodeFlow API (#361) · f370e628
      Da Zheng authored
      * sample layer subgraphs.
      
      * fix.
      
      * fix.
      
      * add layered subgraph.
      
      * fix lint.
      
      * fix.
      
      * fix tutorial.
      
      * fix.
      
      * remove copy_to_parent.
      
      * add num_layers
      
      * move sampling code to sampler.cc
      
      * fix.
      
      * move subgraph construction out.
      
      * Revert "move subgraph construction out."
      
      This reverts commit 24b3d13b0d8ed5f91847ea75a7674ee8f7d45cff.
      
      * change to NodeFlow.
      
      * use NodeFlow in Python.
      
      * use NodeFlowIndex.
      
      * add node_mapping and edge_mapping.
      
      * remove unnecessary code in SSE tutorial.
      
      * Revert "remove unnecessary code in SSE tutorial."
      
      This reverts commit 093f0413d5fa2e63ca5f80c46c80a126a9fb720c.
      
      * fix tutorial.
      
      * move to node_flow.
      
      * update gcn cv updater.
      
      * import NodeFlow.
      
      * update.
      
      * add demo code for vanilla control variate sampler.
      
      * update.
      
      * update.
      
      * add neighbor sampling.
      
      * return flow offsets.
      
      * update node_flow.
      
      * add test.
      
      * fix sampler.
      
      * fix graph index.
      
      * fix a bug in sampler.
      
      * fix map_to_layer_nid and map_to_flow_eid.
      
      * fix apply_flow.
      
      * remove model code.
      
      * implement flow_compute.
      
      * fix a bug.
      
      * reverse the csr physically.
      
      * add mini-batch test.
      
      * add mini batch test.
      
      * update flow_compute.
      
      * add prop_flows
      
      * run on specific nodes.
      
      * test copy
      
      * fix a bug in creating frame in NodeFlow.
      
      * add init gcn_cv_updater.
      
      * fix a minor bug.
      
      * fix gcn_cv_updater.
      
      * fix a bug.
      
      * fix a bug in NodeFlow.
      
      * use new h in gcn_cv_updater.
      
      * add layer_in_degree and layer_out_degree.
      
      * fix gcn_cv_updater for gpu.
      
      * temp fix in NodeFlow for diff context.
      
      * allow enabling/disabling copy back.
      
      * add with-updater option.
      
      * fix a bug in computing degree.
      
      * add with-cv option.
      
      * rename and add comments.
      
      * fix lint complain.
      
      * fix lint.
      
      * avoid assert.
      
      * remove assert.
      
      * fix.
      
      * fix.
      
      * fix.
      
      * fix.
      
      * fix the methods in NodeFlow.
      
      * fix lint.
      
      * update SSE.
      
      * remove gcn_cv_updater.
      
      * correct comments for the schedulers.
      
      * update comment.
      
      * add map_to_nodeflow_nid
      
      * address comment.
      
      * remove duplicated test.
      
      * fix int.
      
      * fix comments.
      
      * fix lint
      
      * fix.
      
      * replace subgraph with NodeFlow.
      
      * move view.
      
      * address comments.
      
      * fix lint.
      
      * fix lint.
      
      * remove static_cast.
      
      * fix docstring.
      
      * fix comments.
      
      * break SampleSubgraph.
      
      * move neighbor sampling to sampler.cc
      
      * fix comments.
      
      * rename.
      
      * split neighbor_list.
      
      * address comments.
      
      * fix.
      
      * remove TODO.
      f370e628
  4. 12 Feb, 2019 1 commit
  5. 17 Dec, 2018 1 commit
  6. 05 Dec, 2018 2 commits
  7. 04 Dec, 2018 1 commit
  8. 03 Dec, 2018 1 commit
  9. 02 Dec, 2018 1 commit
    • Da Zheng's avatar
      [Model] Accelerate SSE (#172) · 57b07fce
      Da Zheng authored
      * 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.
      57b07fce
  10. 20 Nov, 2018 1 commit
    • Da Zheng's avatar
      Fix the convergence problem in SSE. (#162) · 5f89cd66
      Da Zheng authored
      * 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.
      5f89cd66
  11. 14 Nov, 2018 1 commit
    • Da Zheng's avatar
      [API] Subgraph sampling API. (#138) · 2389df81
      Da Zheng authored
      * 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.
      2389df81
  12. 08 Nov, 2018 1 commit
    • Da Zheng's avatar
      [MXNet][API] move to the new API (#123) · bd0e4fa0
      Da Zheng authored
      * 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.
      bd0e4fa0
  13. 26 Oct, 2018 1 commit
    • Da Zheng's avatar
      [GraphIndex] Immutable graph index using MXNet CSRArray (#86) · 9e9a9488
      Da Zheng authored
      * 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.
      9e9a9488