Skip to content

Latest commit

 

History

History
22 lines (13 loc) · 1.47 KB

README.md

File metadata and controls

22 lines (13 loc) · 1.47 KB
Raw

Examples

This folder contains a plethora of examples covering different GNN use-cases. This readme highlights some key examples.

A great and simple example to start with is gcn.py, showing a user how to train a GCN model for node-level prediction on small-scale homogeneous data.

For a simple link prediction example, see link_pred.py.

For examples on Open Graph Benchmark datasets, see the ogbn_*.py examples:

  • ogbn_train.py is an example for training a GNN on the large-scale ogbn-papers100m dataset, containing approximately ~1.6B edges or the medium scale ogbn-products dataset, ~62M edges.
  • ogbn_proteins_deepgcn.py is an example to showcase how to train deep GNNs on the ogbn-proteins dataset.
  • ogbn_papers_100m_cugraph.py shows how to accelerate the ogbn-papers100m workflow using CuGraph.

For examples on using torch.compile, see the examples under examples/compile.

For examples on scaling PyG up via multi-GPUs, see the examples under examples/multi_gpu.

For examples on working with heterogeneous data, see the examples under examples/hetero.

For examples on co-training LLMs with GNNs, see the examples under examples/llm.