Skip to content

Latest commit

 

History

History
41 lines (24 loc) · 1.35 KB

README.md

File metadata and controls

41 lines (24 loc) · 1.35 KB

DEMO

cogdl-trafficPre demo

Usage

API Usage

You can run all kinds of experiments through CogDL APIs, especially experiment. You can also use your own datasets and models for experiments. A quickstart example can be found in the STGCN_example.py. More examples are provided in the examples/.

from cogdl import experiment

basic usage

experiment(dataset="pems-stgcn", model="stgcn")

CogDL Team

CogDL is developed and maintained by Tsinghua, ZJU, BAAI, DAMO Academy, and ZHIPU.AI.

The core development team can be reached at [email protected].

Citing CogDL

Please cite our paper if you find our code or results useful for your research:

@article{cen2021cogdl,
    title={CogDL: Toolkit for Deep Learning on Graphs},
    author={Yukuo Cen and Zhenyu Hou and Yan Wang and Qibin Chen and Yizhen Luo and Xingcheng Yao and Aohan Zeng and Shiguang Guo and Peng Zhang and Guohao Dai and Yu Wang and Chang Zhou and Hongxia Yang and Jie Tang},
    journal={arXiv preprint arXiv:2103.00959},
    year={2021}
}