README.md

---
title: README
emoji: 🏢
colorFrom: purple
colorTo: gray
sdk: static
pinned: false
---

The goal of this repository is to store the different graph datasets currently available as benchmarks, to provide them in an homogeneous and easily loadable way. For example, to load them in PyGeometric, you can do the following:
```python
from datasets import load_dataset

from torch_geometric.data import Data
from torch_geometric.loader import DataLoader

graphs_dataset = load_dataset("graphs-datasets/<dataset-name>")
# For the train set (replace by valid or test as needed)
graphs_list = [Data(graph) for graph in graphs_dataset["train"]]
graphs_pygeometric = DataLoader(graph_list)
```


Currently available graphs are:

<strong> Open Graph Benchmark </strong>, graph classification task: <em> ogbg-molhiv, ogbg-molpcba, ogbg-ppa, ogbg-code2 </em>

- More to come!