文件历史

15 次代码提交

作者 SHA1 备注 提交日期
hgalioulline 6c7c403914 [Feature] Filtered MRR metrics for R-GCN example (#1298)
* Add filtered metrics for R-GCN example

* Add new line to end of file

* Add evaluation protocol argument option for R-GCN example

* Update README

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
2020-03-13 16:25:03 +08:00
Quan (Andy) Gan b05cb84a77 [BUG] Fixes #1262 (#1286) 2020-02-24 13:33:55 +08:00
Zihao Ye 7a80faf120 [Bugfix] Correct the loss function in RGCN model (#1217)
* upd

* upd

* fix

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-02-13 18:35:07 +08:00
Minjie Wang 1c91f460d3 [Hetero][Model] RGCN for heterogeneous input (#885)
* new hetero RGCN

* bgs running

* fix gpu

* am dataset

* fix bug in label preparation

* Fix AM training; add result

* rm sym link

* new embed layer; mutag

* mutag matched; other fix

* minor fix

* dataset refactor

* new data loading

* rm old files

* refactor

* docstring

* include literal nodes in AIFB dataset

* address comments

* docstring
2019-09-29 00:21:17 -04:00
Minjie Wang 708765f0a1 [NN] RGCN modules (#744)
* 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
2019-08-23 16:38:48 -04:00
Sahand 77c58289a7 [Bugfix][Model] Fixing RGCN evaluation bug (#778) 2019-08-20 15:54:03 -04:00
Quan (Andy) Gan 714f2e1ced fix node id dtype (#674) 2019-06-20 10:53:26 +08:00
Giuseppe Futia 2dff1aba83 [Bugfix] Fix shape of test_relation tensor in RGCN (#464) 2019-03-28 14:34:07 -04:00
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