From 32488fc648dc493be2381c0e9b08e0e48869d561 Mon Sep 17 00:00:00 2001 From: Justin Gu Date: Mon, 2 Mar 2026 20:03:00 -0800 Subject: [PATCH 1/3] Add TGB usage to README --- README.md | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 3c973226..898a2bb3 100644 --- a/README.md +++ b/README.md @@ -120,7 +120,7 @@ pip install relbench[ctu] If you use the CTU datasets in your work, please cite [ReDeLEx](https://github.com/jakubpeleska/redelex) as below: -``` +```bibtex @misc{peleska2025redelex, title={REDELEX: A Framework for Relational Deep Learning Exploration}, author={Jakub Peleška and Gustav Šír}, @@ -142,7 +142,7 @@ pip install relbench[dbinfer] ``` If you use the 4DBInfer datasets in your work, please cite [4DBInfer](https://github.com/awslabs/multi-table-benchmark) as below: -``` +```bibtex @article{dbinfer, title={4DBInfer: A 4D Benchmarking Toolbox for Graph-Centric Predictive Modeling on Relational DBs}, author={Wang, Minjie and Gan, Quan and Wipf, David and Cai, Zhenkun and Li, Ning and Tang, Jianheng and Zhang, Yanlin and Zhang, Zizhao and Mao, Zunyao and Song, Yakun and Wang, Yanbo and Li, Jiahang and Zhang, Han and Yang, Guang and Qin, Xiao and Lei, Chuan and Zhang, Muhan and Zhang, Weinan and Faloutsos, Christos and Zhang, Zheng}, @@ -151,6 +151,22 @@ If you use the 4DBInfer datasets in your work, please cite [4DBInfer](https://gi } ``` +**Using TGB datasets** + +[TGB](https://tgb.complexdatalab.com/) datasets and tasks can be loaded in the RelBench format, and they cover bipartite link prediction (`tgbl-*`), heterogeneous link prediction (`thgl-*`), and node property prediction (`tgbn-*`). + +TGB tasks use official TGB evaluation protocols (one-vs-many MRR / Hits@10 for link prediction, NDCG@10 for node property prediction) rather than the standard RelBench metrics. See `relbench/tasks/tgb.py` for the evaluation API. + +For training scripts and the TGB-to-RelBench export pipeline, see the companion repository: [TGB2](https://github.com/pc0618/TGB2). + +If you use the TGB datasets in your work, please cite [TGB](https://tgb.complexdatalab.com/) as below: +```bibtex +@article{huang2023temporal, + title={Temporal Graph Benchmark for Machine Learning on Temporal Graphs}, + journal={NeurIPS Datasets and Benchmarks}, + year={2023} +} +``` # Package Usage From 0af3d14c33ed5c4e45337e859e04790dcf382eaf Mon Sep 17 00:00:00 2001 From: Rishabh Ranjan Date: Mon, 2 Mar 2026 20:57:42 -0800 Subject: [PATCH 2/3] Update README.md --- README.md | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/README.md b/README.md index 898a2bb3..02e3388f 100644 --- a/README.md +++ b/README.md @@ -153,11 +153,7 @@ If you use the 4DBInfer datasets in your work, please cite [4DBInfer](https://gi **Using TGB datasets** -[TGB](https://tgb.complexdatalab.com/) datasets and tasks can be loaded in the RelBench format, and they cover bipartite link prediction (`tgbl-*`), heterogeneous link prediction (`thgl-*`), and node property prediction (`tgbn-*`). - -TGB tasks use official TGB evaluation protocols (one-vs-many MRR / Hits@10 for link prediction, NDCG@10 for node property prediction) rather than the standard RelBench metrics. See `relbench/tasks/tgb.py` for the evaluation API. - -For training scripts and the TGB-to-RelBench export pipeline, see the companion repository: [TGB2](https://github.com/pc0618/TGB2). +[TGB](https://tgb.complexdatalab.com/) datasets and tasks can be loaded in the RelBench format, and they cover bipartite link prediction (`tgbl-*`), heterogeneous link prediction (`thgl-*`), and node property prediction (`tgbn-*`). See `relbench/tasks/tgb.py` for the official TGB evaluation protocols (one-vs-many MRR / Hits@10 for link prediction, NDCG@10 for node property prediction). Standard RelBench style evaluation works as it is. If you use the TGB datasets in your work, please cite [TGB](https://tgb.complexdatalab.com/) as below: ```bibtex From 350075586dd0ffac00ba051fa33c034205c824f3 Mon Sep 17 00:00:00 2001 From: Justin Gu Date: Mon, 2 Mar 2026 21:09:10 -0800 Subject: [PATCH 3/3] Add proper citations for TGB to readme --- README.md | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 02e3388f..c05ee2e5 100644 --- a/README.md +++ b/README.md @@ -158,12 +158,23 @@ If you use the 4DBInfer datasets in your work, please cite [4DBInfer](https://gi If you use the TGB datasets in your work, please cite [TGB](https://tgb.complexdatalab.com/) as below: ```bibtex @article{huang2023temporal, - title={Temporal Graph Benchmark for Machine Learning on Temporal Graphs}, - journal={NeurIPS Datasets and Benchmarks}, + 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} } ``` +```bibtex +@article{huang2024tgb2, + title={TGB 2.0: A Benchmark for Learning on Temporal Knowledge Graphs and Heterogeneous Graphs}, + author={Gastinger, Julia and Huang, Shenyang and Galkin, Mikhail and Loghmani, Erfan and Parviz, Ali and Poursafaei, Farimah and Danovitch, Jacob and Rossi, Emanuele and Koutis, Ioannis and Stuckenschmidt, Heiner and Rabbany, Reihaneh and Rabusseau, Guillaume}, + journal={Advances in Neural Information Processing Systems}, + year={2024} +} +``` + + # Package Usage