文件历史

作者	SHA1	备注	提交日期
Quan (Andy) Gan	4b4186f87d	[Sampler] Change argument type of fanout from list to dict (#1403 )	2020-03-30 15:52:46 +08:00
Mufei Li	0f40c6e49c	[Hetero] Replace card with num_nodes Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>	2020-03-20 16:56:53 +08:00
Minjie Wang	5dd35580f7	[Feature] Improve sampling speed; Better pickle/unpickle; other fixes (#1299 ) * improve performance of sample_neighbors * some more improve * test script * benchmarks * multi process * update more tests * WIP * adding two API for state saving * add create from state * upd test * missing file * wip: pickle/unpickle * more c apis * find the problem of empty data array * add null array; pickling speed is bad * still bad perf * still bad perf * wip * fix the pickle speed test; now everything looks good * minor fix * bugfix * some lint fix * address comments * more fix * fix lint * add utest for random.choice * add utest for dgl.rand_graph * fix cpp utests * try fix ci * fix bug in TF backend * upd choice docstring * address comments * upd * try fix compile * add comment	2020-03-02 23:16:05 +08:00
Quan (Andy) Gan	c3a33407fe	[Feature] PinSAGE sampler (#1249 ) * [WIP] PinSAGE operators * moved the edge remapping mess into C * some docstrings * lint * lint x2 * lint x3 * skip gpu test on topk * extend pinsage to any metapath * lint x4 * addresses #1265 * add always_preserve (fixes #1266) and fix a silly bug * disable gpu test on compaction * lint * fix a horrible bug and add more tests * lint * addresses comments * lint * bugfix * addresses comments Co-authored-by: Minjie Wang <minjie.wang@nyu.edu>	2020-02-28 20:46:13 +08:00
Minjie Wang	f8d4264e5b	[Feature] Neighbor-hood based sampling APIs (#1251 ) * WIP: working on random choices * light slice * basic CPU impl * add python binding; fix CreateFromCOO and CreateFromCSR returning unitgraph * simple test case works * fix bug in slicing probability array * fix bug in getting the correct relation graph * fix bug in creating placeholder graph * enable omp * add cpp test * sample topk * add in\|out_subgraph * try fix lint; passed all unittests * fix lint * fix msvc compile; add sorted flag and constructors * fix msvc * coosort * COOSort; CSRRowWiseSampling; CSRRowWiseTopk * WIP: remove DType in CSR and COO; Restrict data array to be IdArray * fix all CSR ops for missing data array * compiled * passed tests * lint * test sampling out edge * test different per-relation fanout/k values * fix bug in random choice * finished cpptest * fix compile * Add induced edges * add check * fixed bug in sampling on hypersparse graph; add tests * add ascending flag * in\|out_subgraph returns subgraph and induced eid * address comments * lint * fix	2020-02-17 21:25:21 +08:00
Quan (Andy) Gan	a9dabcc769	[Feature] Random Walk for 0.5 (#1209 ) * trying to refactor IndexSelect * partial implementation * add index select and assign for floats as well * move to random choice source * more updates * fixes * fixes * more fixes * adding python impl * fixes * unit test * lint * lint x2 * lint x3 * update metapath2vec * debugging performance * still debugging for performance * tuning * switching to succvec * redo * revert non-uniform sampler to use vector * still not fast * why does this crash with OpenMP??? * because there was a data race!!! * add documentations and remove assign op * lint * lint x2 * lol what have i done * lint x3 * fix and disable gpu testing * bugfix * generic random walk * reorg the random walk source code * Update randomwalks.h * Update randomwalks_cpu.cc * rename file * move internal function to anonymous ns * reorg & docstrings * constant restart probability * docstring fix * more commit * random walk with restart, tested * some fixes * switch to using NDArray for choice * massive fix & docstring * lint x? * lint x?? * fix * export symbols * skip gpu test * addresses comments * replaces another VecToIdArray * add randomwalks.h to include * replace void * with template	2020-01-31 19:52:08 +08:00

1 2

56 次代码提交