Skip to content

cvignac/graph_regression_template

Repository files navigation

A template for graph-level regression, demonstrated on QM9.

The network implemented is a graph attention network.

The code uses the following tools:

  • pytorch_geometric
  • pytorch_lightning
  • hydra
  • weights and biases

Installation

  • First follow the installation guide of pytorch geometric.
  • Install the other packages with pip install -r requirements.txt

author: Clement Vignac

About

A small model that performs graph regression on QM9

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages