Add ECHOBenchmark dataset#10648
Conversation
for more information, see https://pre-commit.ci
…ytorch_geometric into add-echo-dataset
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
…ytorch_geometric into add-echo-dataset
for more information, see https://pre-commit.ci
|
Hi, we’ve addressed the issues raised in the previous review and updated the PR accordingly. If possible, we would greatly appreciate a review and approval before the ICLR 2026 conference. Thank you for your time! |
|
Hi @rusty1s , @wsad1 , and @akihironitta I wanted to follow up on this pull request regarding the integration of our new benchmark for evaluating GNNs' long-range capabilities. This work has been published at ICLR 2026. Please let us know if you have any feedback or if there are any specific updates required on our end to move forward. Thanks, |
|
Hi @rusty1s, @wsad1, and @akihironitta, I wanted to kindly follow up on this PR. We have kept the branch up to date with This benchmark was accepted at ICLR 2026, and we would be grateful for any feedback, requested changes, or an approval if everything looks good from your side. Please let us know if there is anything else we should address to help move this forward. Thanks again for your time and for maintaining PyG! |
Summary
This PR adds the ECHOBenchmark dataset, introduced in:
The dataset consists of 5 tasks:
Each task provides train/val/test splits as lists of
torch_geometric.data.Data.Features
InMemoryDatasetChecklist
torch_geometric/datasets/echo_benchmark.pyhomo_datasetsintorch_geometric/datasets/__init__.pyexamples/echo_benchmark.pytest/datasets/test_echo_benchmark.py