CGCNN2
As the original Crystal Graph Convolutional Neural Networks (CGCNN) repository is no longer actively maintained, this repository is a reproduction of CGCNN by Xie et al. It includes necessary updates for deprecated components and a few additional functions to ensure smooth operation. Despite its age, CGCNN remains a straightforward and fast deep learning framework that is easy to learn and use.
The package provides following major functions:
- Training a CGCNN model with a customized dataset.
- Predicting material properties with a pre-trained CGCNN model.
- Fine-tuning a pre-trained CGCNN model on a new dataset.
- Extracting atomic features as descriptors for the downstream task.
Installation
Make sure you have a Python interpreter, preferably version 3.10 or higher. Then, you can simply install xdatbus from
PyPI using pip:
pip install cgcnn2
If you'd like to use the latest unreleased version on the main branch, you can install it directly from GitHub:
pip install git+https://github.com/jcwang587/cgcnn2
Get Started
cgcnn-ft --help
References
The original paper describes the details of the CGCNN framework:
@article{PhysRevLett2018,
title = {Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties},
author = {Xie, Tian and Grossman, Jeffrey C.},
journal = {Phys. Rev. Lett.},
volume = {120},
issue = {14},
pages = {145301},
numpages = {6},
year = {2018},
month = {Apr},
publisher = {American Physical Society},
doi = {10.1103/PhysRevLett.120.145301},
url = {https://link.aps.org/doi/10.1103/PhysRevLett.120.145301}
}
Metadata
Release files for cgcnn2 0.3.2
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| cgcnn2-0.3.2.tar.gz | 15.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| cgcnn2-0.3.2-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 31.2 kB
Release files / cgcnn2-0.3.2.tar.gz
| Download URL | cgcnn2-0.3.2.tar.gz |
|---|---|
| Size | 15.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
824733fc61aa3295844aa84a4b67fcf5b8942aa5e72a534ff891a9380b83e43d
|
|
BLAKE2b-256 checksum How to use checksums |
fbfac5ef11d1418520c4982a1ac4fe1e059a15b62f1103c3ea77404153812e29
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/6.1.0 CPython/3.12.9
|
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 Mar 19, 2025.
Transparency logRelease files / cgcnn2-0.3.2-py3-none-any.whl
| Download URL | cgcnn2-0.3.2-py3-none-any.whl |
|---|---|
| Size | 16.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
3344e28812588b638703012cc0051c2d0d9b6169a11a934a4ca204560aebd404
|
|
BLAKE2b-256 checksum How to use checksums |
13fd9cccbc81278f1bfbc6eed6702236760cfc18c4585ca49e89e124ac0f5472
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/6.1.0 CPython/3.12.9
|
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 Mar 19, 2025.
Transparency log