Temporal Graph Benchmark for Machine Learning on Temporal Graphs (NeurIPS 2023 Datasets and Benchmarks Track)
TGB 2.0: A Benchmark for Learning on Temporal Knowledge Graphs and Heterogeneous Graphs (preprint)
Overview of the Temporal Graph Benchmark (TGB) pipeline:
numpy
, PyTorch
and PyG compatible TemporalData
formats. To submit to TGB leaderboard, please fill in this google form
See all version differences and update notes here
Excited to announce TGB 2.0, expanding TGB to Temporal Knowledge Graphs and Temporal Heterogeneous Graphs
See our preprint here for details. Please install locally first. We welcome your feedback and suggestions.
Excited to announce TGX, a companion package for analyzing temporal graphs in WSDM 2024 Demo Track
TGX supports all TGB datasets and provides numerous temporal graph visualization plots and statistics out of the box. See our paper: Temporal Graph Analysis with TGX and TGX website.
Excited to announce that TGB has been accepted to NeurIPS 2023 Datasets and Benchmarks Track
Thanks to everyone for your help in improving TGB! we will continue to improve TGB based on your feedback and suggestions.
Please update to version 0.9.2
0.9.2
Update the fix for tgbl-flight
where now the unix timestamps are provided directly in the dataset. If you had issues with tgbl-flight
, please remove TGB/tgb/datasets/tgbl_flight
and redownload the dataset for a clean install
You can install TGB via pip. Requires python >= 3.9
pip install py-tgb
The project website can be found here.
The API documentations can be found here.
all dataset download links can be found at info.py
TGB dataloader will also automatically download the dataset as well as the negative samples for the link property prediction datasets.
if website is unaccessible, please use this link instead.
examples/linkproppred
folder for example scripts to run TGN, DyRep and EdgeBank on TGB datasets.examples/nodeproppred
folder for example scripts to run TGN, DyRep and EdgeBank on TGB datasets.We thank the OGB team for their support throughout this project and sharing their website code for the construction of TGB website.
If code or data from this repo is useful for your project, please consider citing our paper:
@article{huang2023temporal,
title={Temporal graph benchmark for machine learning on temporal graphs},
author={Huang, Shenyang and Poursafaei, Farimah and Danovitch, Jacob and Fey, Matthias and Hu, Weihua and Rossi, Emanuele and Leskovec, Jure and Bronstein, Michael and Rabusseau, Guillaume and Rabbany, Reihaneh},
journal={Advances in Neural Information Processing Systems},
year={2023}
}