What is CogDL?
CogDL is a graph representation learning toolkit that allows researchers and developers to easily train and compare baseline or custom models for node classification, link prediction and other tasks on graphs. It provides implementations of many popular models, including: non-GNN Baselines like Deepwalk, LINE, NetMF, GNN Baselines like GCN, GAT, GraphSAGE.
|
Task-OrientedCogDL focuses on tasks on graphs and provides corresponding models, datasets, and leaderboards.
Multiple TasksCogDL supports node classification and link prediction tasks on homogeneous / heterogeneous networks, as well as graph classification.
|
Easy-RunningCogDL supports running multiple experiments simultaneously on multiple models and datasets under a specific task using multiple GPUs.
ExtensibilityYou can easily add new datasets, models and tasks and conduct experiments for them!
|