THUDM / cogdl / 358

76%

push

travis-ci-com

[Framework] Unify the GNN training loop (#285)

* new framework

* Update runner/wrappers and models

* Add link_prediction wrappers

* Support multiplex embedding

* Add gcc

* support ddp

* Add heterogeneous wrappers

* update Evaluator

* update

* Fix experiment API

* Add loggers

* Fix experiment API

* Fix bugs in ddp_training

* fix bugs

* Fix hyperparameter search

* Fix hyperparameter search

* Remove tasks/trainers & Fix models

* [BugFix] Fix bugs in graph_classification datawrapper

* modify custom_dataset

* fix bugs for graph_classification

* Fix some tests

* [BugFix] Fix bugs in link prediction and graph classification

* modify custom_dataset

* fix bugs for graph_classification

* fix bugs in defining custom dataset

* fix bugs in link_prediction and graph_classification

* Fix some tests

* [BugFix] fix bugs in gnn_link_prediction

* modify custom_dataset

* fix bugs for graph_classification

* fix bugs in defining custom dataset

* fix bugs in link_prediction and graph_classification

* fix bugs in gnn_link_prediction

* Fix tests & examples

* Fix tests

* Try to fix metaclass

* Fix GenericMeta

* Fix tests

* Fix tests

* Improve coverage

* Improve coverage

* Python 3.7

* fix bugs in sagn and pprgo tests

* Change to trainer

Co-authored-by: THINK2TRY <48679723+THINK2TRY@users.noreply.github.com>
Co-authored-by: THINK2TRY <hzyjuly11@163.com>

4607 of 4607 new or added lines in 135 files covered. (100.0%)

12949 of 16948 relevant lines covered (76.4%)

0.76 hits per line