• 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
graph.py 98.5 KB