* 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
* [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>
* Several optimizations on DGL-KG:
1. Sorted positive edges for sampling which can reduce random
memory access during positive sampling
2. Asynchronous node embedding update
3. Balanced Relation Partition that gives balanced number of
edges in each partition. When there is no cross partition
relation, relation embedding can be pin into GPU memory
4. tunable neg_sample_size instead of fixed neg_sample_size
* Fix test
* Fix test and eval.py
* Now TransR is OK
* Fix single GPU with mix_cpu_gpu
* Add app tests
* Fix test script
* fix mxnet
* Fix sample
* Add docstrings
* Fix
* Default value for num_workers
* Upd
* upd
* graph format
* fix lint
* lint
* fix
* unit test
* lint
* add magic num
* move serialize out of struct
* lint
Co-authored-by: zhoujinjing09 <zhoujinjing09@users.noreply.github.com>
* unit graph that prefers coo queries
* auto detect coo preference
* forgot some functions
* disable lint on detect_prefer_coo
* reorg
* change comment
* lint
* fix
* move array_utils.h to src
* compact graph impl
* fix redundant copying in idhashmap
* docstring
* moving preference detection to C
* lint
* fix unit test & address comments
* hypersparse autorestrict
* docstring & fix
* revert copyto and asnumbits
* fix stupid bug
* lint
* leave a TODO for sorted COO
* fixing same node type mapping to different id in different graphs
* addresses comments
* made induced nodes a feautre column
* lint?
* 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
* Add boundary check for heterograph build with card as input
* Fix when u or v is empty
* fix test_kernel.py error print
* Revert "fix test_kernel.py error print"
This reverts commit a71c20292549c0ac62a5326c30669ca4bde8febc.
* Turn op validation check on graph and bipartite by default
* upd
* udp
* upd
* update test
* Add weight based edge sampler
* Can run, edge weight work.
TODO: test node weight
* Fix node weight sample
* Fix y
* Update doc
* Fix syntex
* Fix
* Fix GPU test for sampler
* Fix test
* Fix
* Refactor EdgeSampler to act as class object not function that it
can record its own private states.
* clean
* Fix
* Fix
* Fix run bug on kg app
* update
* update test
* test
* Simply python API and fix some C code
* Fix
* Fix
* Fix syntex
* Fix
* Update API description
* add replacement for edge sampler
* Now edge sampler support replacement and no-replacement
* Fix
* Fix
* change kg/app to use edge sampler with replacement config
* Update replacement algo
* Fix syntax
* Update
* Update
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
* Add weight based edge sampler
* Can run, edge weight work.
TODO: test node weight
* Fix node weight sample
* Fix y
* Update doc
* Fix syntex
* Fix
* Fix GPU test for sampler
* Fix test
* Fix
* Refactor EdgeSampler to act as class object not function that it
can record its own private states.
* clean
* Fix
* Fix
* Fix run bug on kg app
* update
* update test
* test
* Simply python API and fix some C code
* Fix
* Fix
* Fix syntex
* Fix
* Update API description
Zihao Ye