文件历史

7 次代码提交

作者 SHA1 备注 提交日期
Lingfan Yu d3c24cc25b [BugFix] Fix bug in RGCN data processing and use index_select to improve speed (#429)
* use index_select instead of __getitem__

* fix bug in dataset processing

* fix edge_type shape bug

* comments
2019-03-04 10:52:21 -05:00
Zihao Ye ae1806f651 [Bugfix] RGCN tutorial (#388)
* fix rgcn tutorial

* small fix

* upd
2019-02-15 11:14:56 +08:00
Giuseppe Futia 6c3dba867a [Tutorial] Update prerequisites of README (#380)
* Update prerequisites of README

* dependencies for pytorch models

* dependencies for mxnet models

* minor
2019-02-11 22:58:17 -05:00
brett koonce 707334ce85 minor spelling tweaks (#349)
* minor spelling tweaks

* Update CONTRIBUTORS.md
2019-01-10 12:02:31 -05:00
Gan Quan ac660f45ef [Doc] Readme (#265)
* [Doc] updated readme

* contribution guide change

* small edits

* add perf #

* Update README.md

* Update README.md

* add others...

* rewrite scalability

* model & code links

* some note for treelstm

* rewrite scalability
2018-12-06 00:31:04 -05:00
Minjie Wang e557ed89ff [Doc][Model] example folder doc and model summary (#259)
* update example readme

* mx example readme

* add results in readme

* mx rgcn readme
2018-12-05 14:35:14 -05:00
Lingfan Yu b98dc92c59 [Model] Relational GCN (#55)
* 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
2018-12-03 13:01:27 -05:00