- 23 Aug, 2019 1 commit
-
-
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
-
- 05 Dec, 2018 1 commit
-
-
Yizhi Liu authored
* entity classify work for examples * add loop_msg * remove wrong assert * remove one reshape * add readme * add MRR * remove mrr from entity task
-
- 03 Dec, 2018 1 commit
-
-
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
-