Skip to main content

DOI License: MIT Python package PyPI version

GAME-Net-UQ

This repository contains the Python code used to train and evaluate GAME-Net-UQ, a graph neural network with uncertainty quantification (UQ) for predicting the DFT energy of relaxed species and transition states adsorbed on metal surfaces.

Install

pip install gamenet-uq

The main dependencies of the repo can be found in pyproject.toml

DFT dataset

The DFT dataset fg.db (217 MB) used to train the GNN will be soon uploaded to Zenodo as ASE database including the DFT VASP relaxed geometries, simulation settings, and other metadata.

Graph dataset generation from ASE databases

The graph dataset (92 MB) can be automatically generated from the ASE database with the script gen_dataset.py. The same script can be used to generate your custom dataset from external ASE databases.

Model training and finetuning

To train the model, run the script train_mve.py. The input template file provides an explanation for each entry required in the training configuration file.

python train_mve.py -i input.toml -o output_dirname

Pretrained model

The final pretrained model can be employed with CARE (link).

License

The code is released under the MIT license.

Reference

  • A Foundational Model for Reaction Networks on Metal Surfaces
    Authors: S. Morandi, O. Loveday, T. Renningholtz, S. Pablo-García, R. A. Vargas Hernáńdez, R. R. Seemakurthi, P. Sanz Berman, R. García-Muelas, A. Aspuru-Guzik, and N. López
    DOI: 10.26434/chemrxiv-2024-bfv3d

Metadata

Release files for gamenet-uq 0.1.3

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for gamenet-uq 0.1.3
File Size Uploaded Provenance Actions
gamenet_uq-0.1.3.tar.gz 42.3 kB With provenance Details for gamenet_uq-0.1.3.tar.gz

Built distribution (wheel)

Table of built distributions (wheels) for gamenet-uq 0.1.3
File Interpreter ABI Platform Actions
gamenet_uq-0.1.3-py3-none-any.whl Python 3 none any Details for gamenet_uq-0.1.3-py3-none-any.whl

Total release size: 87.3 kB

Release files / gamenet_uq-0.1.3.tar.gz

Download URL gamenet_uq-0.1.3.tar.gz
Size 42.3 kB
Tags Source
SHA-256 checksum
How to use checksums
7650a0eecdbadaf44031a4b74ff4d501fb9568568f66551527ef0ae0aebaae85
BLAKE2b-256 checksum
How to use checksums
a939573cefe3b6dc516719bb0673832b538d76a5328c401f3de63ffcf58ed549
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/6.1.0 CPython/3.13.7

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Nov 18, 2025.

Transparency log

Release files / gamenet_uq-0.1.3-py3-none-any.whl

Download URL gamenet_uq-0.1.3-py3-none-any.whl
Size 45.0 kB
Tags Python 3
SHA-256 checksum
How to use checksums
7b5ca2904d89ef82461d7a900b6dd0b748ea2d67e0dd1bdb1590c0876efa070e
BLAKE2b-256 checksum
How to use checksums
e5169110115868b09cb295717ddd15a5b8675c811e15ee0077ed7420846705da
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/6.1.0 CPython/3.13.7

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Nov 18, 2025.

Transparency log

Security for gamenet-uq 0.1.3

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

Full provenance

All 2 files in this release have been uploaded with provenance metadata.

Select from 2 files

Attestation for gamenet_uq-0.1.3.tar.gz

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Nov 18, 2025.

Transparency log

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page