1. 30 Jul, 2020 1 commit
  2. 13 Feb, 2020 1 commit
  3. 23 Aug, 2019 1 commit
    • Minjie Wang's avatar
      [NN] RGCN modules (#744) · 708765f0
      Minjie Wang authored
      * rgcn module
      
      * support id input
      
      * WIP: model codes
      
      * use faster index select
      
      * dropout
      
      * self loop
      
      * WIP: link prediction
      
      * fix lint
      
      * WIP: docs
      
      * docstring
      
      * docstring
      
      * merge two child classes
      
      * mxnet rgcn module
      
      * fix lint
      
      * fix lint
      
      * fix rename bug
      
      * add uniform edge sampler
      
      * fix fn name
      
      * docstring
      
      * fix mxnet rgcn module
      
      * fix mx rgcn
      
      * enable test on cuda
      708765f0
  4. 05 Dec, 2018 1 commit
  5. 03 Dec, 2018 1 commit
    • Lingfan Yu's avatar
      [Model] Relational GCN (#55) · b98dc92c
      Lingfan Yu authored
      * data preprocessing for rgcn
      
      * edge subgraph
      
      * WIP: RGCN
      
      * use edge feature in spmv
      
      * fix bugs
      
      * match AIFB accuracy
      
      * match mutag accuracy
      
      * avoid materializing in featureless case
      
      * remove untouched nodes and relabel nodes
      
      * fix python list concatenate overhead
      
      * sparsely store edge types
      
      * refactor entity classify code for clean link prediction implementation
      
      * further refactor code
      
      * refactoring
      
      * rgcn block decompose layers
      
      * link predict dataset
      
      * link predict model and eval code
      
      * dropout, self-loop, regularization, etc, plus bug fixes
      
      * update to new api
      
      * dataset update
      
      * bugs, WIP, need to impl early stopping and filtered metrics
      
      * instruction to run, and minor
      
      * group conv and early stop
      
      * clean slow code
      
      * some code comments
      
      * use new api in model code
      
      * change data preprocessing
      
      * entity classify model
      
      * WIP
      
      * move dgl graph out of model
      
      * hot fix for extract zip
      
      * fix link predict model
      
      * use latest dgl apis
      
      * still have memory issue...
      
      * bug fix and move inference to cpu
      
      * move rgcn data processing to contrib
      
      * th.allclose -> U.allclose
      
      * minor change in readme
      
      * fix memory issue in entity classify
      
      * fix and testing code for link predict
      
      * fix entity classify
      
      * clean up
      
      * fix comments
      
      * revert erroneous git merge changes
      
      * code clean up and more comments
      
      * minor
      
      * dependent package version
      b98dc92c