• Minjie Wang's avatar
    [Feature] Neighbor-hood based sampling APIs (#1251) · f8d4264e
    Minjie Wang authored
    * 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
    f8d4264e
rowwise_topk.cc 3.05 KB