1. 01 Jul, 2019 1 commit
    • HQ's avatar
      [Model] DiffPool with both DGL and tensorized operatons (#665) · 684a61ad
      HQ authored
      * diffpool original file added
      
      * make diffpool fuse up and running
      
      * minor tweak on tu dataset statistics method
      
      * fix tu
      
      * break
      
      * delete break
      
      * pre_org
      
      * diffpool fuse reorg
      
      * fix random shuffling
      
      * fix bn
      
      * add dgl layers
      
      * early stopping
      
      * add readme
      
      * fix
      
      * add diffpool preprocess script
      
      * tweak tu dataset
      
      * tweak
      
      * tweak
      
      * tweak
      
      * tweak
      
      * tweak
      
      * preprocess dataset
      
      * fix early stopping
      
      * fix
      
      * fix
      
      * fix
      
      * tweak
      
      * readme
      
      * code review
      
      * code review
      
      * dataset code review
      
      * update README
      
      * code review
      
      * tu doc
      684a61ad
  2. 29 Mar, 2019 1 commit
    • 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
  3. 15 Jun, 2018 1 commit
  4. 14 Jun, 2018 1 commit