• Quan (Andy) Gan's avatar
    [Model][Feature] PinSage & Random Walk with Restart (#453) · 6066fee9
    Quan (Andy) Gan authored
    * random walk traces generation
    
    * remove outdated comments
    
    * oops put in the wrong place
    
    * explicit inline
    
    * moving rand_r to util
    
    * pinsage-like model on movielens
    
    * the code runs now
    
    * support cuda
    
    * using readonly graph
    
    * moving random walk to public function
    
    * per-thread seed and openmp support
    
    * pinsage-like model on movielens
    
    * the code runs now
    
    * support cuda
    
    * using readonly graph
    
    * using C random walk
    
    * removing profile decorators
    
    * param initialization
    
    * no grad
    
    * leaky relu fixes everything
    
    * train and save
    
    * WIP
    
    * WIP
    
    * WIP
    
    * seems to work
    
    * evaluation output
    
    * swapping order of val/test and train
    
    * debug
    
    * hyperparam tuning
    
    * prior/training dataset split changes
    
    * random walk reorg
    
    * random walk with restart
    
    * signed comparison fix
    
    * migrating random walk to nodeflow
    
    * Revert "migrating random walk to nodeflow"
    
    This reverts commit f2565347cced7c912a58a529b257c033d9f375b7.
    
    * add README and remove dataset
    
    * new endpoint
    
    * lint
    
    * lint x2
    
    * oops forgot test
    
    * including bpr - better for baseline
    
    * addressing fixes
    
    * throwing random walks out from SamplerOp class
    
    * forgot to move RandomWalk; why did this even work?
    
    * removing legacy garbage
    
    * add todo
    
    * address comments
    
    * stupid bug fix
    
    * call ndarrayvector converter to handle traces
    6066fee9
graph_apis.cc 16.5 KB