CogDL v0.1.2

cenyk1230 released this 17 Nov 15:32

04662b2

New Features

#48 Support the pre-training task on molecule graphs
#38 Add Trainer API for flexible training
#38 Add Sampler API for training large-scale datasets and now supports GraphSAINT sampler.

New Models

#48 STP-GNN for pre-training
#38 GPT-GNN for node classification
#39 Triple based knowledge embedding methods (complex, distmult, rotate, transe)
#48 DeeperGCN for node classification
#48 GCNII for node classification

New Results

#51 Update the leaderboard of the unsupervised node classification task
#48 Update the leaderboard of the semi-supervised node classification task
#48 Update the leaderboard of the graph classification task

New Datasets

#50 Add some molecule datasets
- "bio" and "chem" in Jure's paper.
- BBBP and BACE
#38 Add OGB datasets

New Examples

#51 Add many examples of embedding methods
#48 Add many examples of graph neural networks

Requirement Update

#38 CogDL now requires ogb

Miscellaneous

#50 #54 Remove saved/ folder and support downloading pre-trained GCC model
#52 Improve the coverage to 80%

Assets 2