Da Zheng
|
fc7775a20f
|
Define node/edge Ids in NodeFlow more clearly (#628)
* add tests.
* distinguish layer-local nid and nodeflow nid.
* use numpy assert_array_equal and assert_allclose
* fix map_from_parent_nid
* fix test
* fix test.
* renmae remap.
* update doc.
* update doc.
* update doc.
* fix test.
* fix test.
|
2019-06-09 12:10:07 -07:00 |
|
Da Zheng
|
b2b8be25bb
|
[API] update graph store API. (#549)
* add init_ndata and init_edata in DGLGraph.
* adjust SharedMemoryGraph API.
* print warning.
* fix comment.
* update example
* fix.
* fix examples.
* add unit tests.
* add comments.
|
2019-05-21 03:27:05 -07:00 |
|
Da Zheng
|
cdfca992e1
|
[BUGFix] Improve multi-processing training (#526)
* fix.
* add comment.
* remove.
* temp fix.
* initialize for shared memory.
* fix graphsage.
* fix gcn.
* add more unit tests.
* add more tests.
* avoid creating shared-memory exclusively.
* redefine remote initializer.
* improve initializer.
* fix unit test.
* fix lint.
* fix lint.
* initialize data in the graph store server properly.
* fix test.
* fix test.
* fix test.
* small fix.
* add comments.
* cleanup server.
* test graph store with a random port.
* print.
* print to stderr.
* test1
* test2
* remove comment.
* adjust the initializer signature.
|
2019-05-20 14:23:19 -07:00 |
|
Da Zheng
|
3a1392e641
|
[Model] add multiprocessing training with sampling. (#484)
* reorganize sampling code.
* add multi-process training.
* speed up gcn_cv
* fix graphsage_cv.
* add new API in graph store.
* update barrier impl.
* support both local and distributed training.
* fix multiprocess train.
* fix.
* fix barrier.
* add script for loading data.
* multiprocessing sampling.
* accel training.
* replace pull with spmv for speedup.
* nodeflow copy from parent with context.
* enable GPU.
* fix a bug in graph store.
* enable multi-GPU training.
* fix lint.
* add comments.
* rename to run_store_server.py
* fix gcn_cv.
* fix a minor bug in sampler.
* handle error better in graph store.
* improve graphsage_cv for distributed mode.
* update README.
* fix.
* update.
|
2019-05-06 23:51:56 -07:00 |
|
Da Zheng
|
6124667f42
|
[Model] Refactor the sampling examples (#498)
* reorganize sampling code.
* speedup gcn_ns.
* speed up gcn_cv
* fix graphsage_cv.
* undo the modification.
* accel training.
* update readme.
|
2019-04-17 01:31:14 -07:00 |
|
Minjie Wang
|
ca2a7e1ca1
|
[Refactor] Nodeflow, sampling, CAPI (#430)
* enable cython
* add helper function and data structure for void_p vector return
* move sampler from graph index to contrib.sampling
* WIP
* WIP
* refactor layer sampling
* pass tests
* fix lint
* fix graphsage
* remove comments
* pickle test
* fix comments
* update dev guide for cython build
|
2019-03-05 14:07:22 -05:00 |
|
Da Zheng
|
bea07b41b3
|
[Sampler] fix the API of neighbor sampler. (#407)
* don't return aux_info.
* fix sampler test.
* fix sse.
* fix.
* add comment.
|
2019-02-28 13:45:21 -08:00 |
|
Ziyue Huang
|
7e30382e4f
|
[Model][MXNet] neighbor sampling & skip connection & control variate & graphsage (#322)
* neighbor sampling draft
* val/test acc
* control variate draft
* control variate
* update
* fix new_history
* maintain aggregated history while updating new history
* preprocess the first layer, change push to pull
* update
* fix subg_degree
* nodeflow
* clear
* readme
* doc and unittest for self loop
* address comments
* rename
* update
* fix
* Update node_flow.py
* Update node_flow.py
|
2019-02-28 13:29:51 -08:00 |
|