cgcnn

所属分类:人工智能/神经网络/深度学习
开发工具:Python
文件大小:6011KB
下载次数:1
上传日期:2021-09-06 05:23:51
上 传 者sh-1993
说明:  用于预测材料特性的晶体图卷积神经网络。
(cgcnn,Crystal graph convolutional neural networks for predicting material properties.)

文件列表:
LICENSE (1065, 2021-09-06)
cgcnn (0, 2021-09-06)
cgcnn\__init__.py (0, 2021-09-06)
cgcnn\data.py (12708, 2021-09-06)
cgcnn\model.py (6655, 2021-09-06)
data (0, 2021-09-06)
data\material-data (0, 2021-09-06)
data\material-data\mp-ids-27430.csv (291980, 2021-09-06)
data\material-data\mp-ids-3402.csv (33332, 2021-09-06)
data\material-data\mp-ids-46744.csv (494045, 2021-09-06)
data\sample-classification (0, 2021-09-06)
data\sample-classification\1000041.cif (4568, 2021-09-06)
data\sample-classification\1000050.cif (2416, 2021-09-06)
data\sample-classification\1101051.cif (2407, 2021-09-06)
data\sample-classification\1507756.cif (2389, 2021-09-06)
data\sample-classification\7206075.cif (3286, 2021-09-06)
data\sample-classification\9000046.cif (2183, 2021-09-06)
data\sample-classification\9009743.cif (2303, 2021-09-06)
data\sample-classification\9011050.cif (4985, 2021-09-06)
data\sample-classification\9011998.cif (5030, 2021-09-06)
data\sample-classification\9012304.cif (5080, 2021-09-06)
data\sample-classification\atom_init.json (28392, 2021-09-06)
data\sample-classification\id_prop.csv (100, 2021-09-06)
data\sample-regression (0, 2021-09-06)
data\sample-regression\1000041.cif (4568, 2021-09-06)
data\sample-regression\1000050.cif (2416, 2021-09-06)
data\sample-regression\1101051.cif (2407, 2021-09-06)
data\sample-regression\1507756.cif (2389, 2021-09-06)
data\sample-regression\7206075.cif (3286, 2021-09-06)
data\sample-regression\9000046.cif (2183, 2021-09-06)
data\sample-regression\9009743.cif (2303, 2021-09-06)
data\sample-regression\9011050.cif (4985, 2021-09-06)
data\sample-regression\9011998.cif (5030, 2021-09-06)
data\sample-regression\9012304.cif (5080, 2021-09-06)
data\sample-regression\atom_init.json (28392, 2021-09-06)
data\sample-regression\id_prop.csv (121, 2021-09-06)
... ...

# Pre-trained CGCNN models This directory includes several pre-trained CGCNN models that one can use to predict the material properties of new crystals. We encourage users to report their own CGCNN models for other properties which might benefit the community. ## Pre-trained models ### Regression | File | Property | Units | Data Ref. | Model Ref. | | --------------------------- | ---------------- | -------- | ------------------------------------------------------------ | ------------------------------------------------------------ | | `formation-energy-per-atom` | Formation Energy | eV/atom | [Jain et al.](https://aip.scitation.org/doi/10.1063/1.4812323) | [Xie et al.](https://link.aps.org/doi/10.1103/PhysRevLett.120.145301) | | ` final-energy-per-atom` | Absolute Energy | eV/atom | [Jain et al.](https://aip.scitation.org/doi/10.1063/1.4812323) | [Xie et al.](https://link.aps.org/doi/10.1103/PhysRevLett.120.145301) | | `band-gap` | Band Gap | eV | [Jain et al.](https://aip.scitation.org/doi/10.1063/1.4812323) | [Xie et al.](https://link.aps.org/doi/10.1103/PhysRevLett.120.145301) | | `efermi` | Fermi Energy | eV/atom | [Jain et al.](https://aip.scitation.org/doi/10.1063/1.4812323) | [Xie et al.](https://link.aps.org/doi/10.1103/PhysRevLett.120.145301) | | `bulk-moduli` | Bulk Moduli | log(GPa) | [Jain et al.](https://aip.scitation.org/doi/10.1063/1.4812323) | [Xie et al.](https://link.aps.org/doi/10.1103/PhysRevLett.120.145301) | | `shear-moduli` | Shear Moduli | log(GPa) | [Jain et al.](https://aip.scitation.org/doi/10.1063/1.4812323) | [Xie et al.](https://link.aps.org/doi/10.1103/PhysRevLett.120.145301) | | `poisson-ratio` | Poisson Ratio | ” | [Jain et al.](https://aip.scitation.org/doi/10.1063/1.4812323) | [Xie et al.](https://link.aps.org/doi/10.1103/PhysRevLett.120.145301) | ### Classification | File | Positive | Negative | Data Ref. | Model Ref. | | --------------------------- | -------- | ------------- | ------------------------------------------------------------ | ------------------------------------------------------------ | | `semi-metal-classification` | Metal | Semiconductor | [Jain et al.](https://aip.scitation.org/doi/10.1063/1.4812323) | [Xie et al.](https://link.aps.org/doi/10.1103/PhysRevLett.120.145301) | ## Before using pre-trained models - CGCNN models (and machine learning models in general) can only generalize to crystals from the same ditribution as training data. It is adviced to check **Data Ref.** before using pre-trained models. For instance, Materials Project uses [ICSD](https://icsd.fiz-karlsruhe.de/search/index.xhtml;jsessionid=E3291AF7E25ED34B31B9AD5A9CBF80A1) structures as input, which includes experimentally synthesized crystal structures. Significant errors can be expected if a CGCNN model trained on Materials Project is used to predict the properties of imaginary, thermadynamically unstable crystals. - CGCNN models have prediction errors. It is adviced to check **Model Ref.** to understand their accuracy before using pre-trained models. ## How to cite If you used any pre-trained models, please cite both **Data Ref.** and **Model Ref.** because data and model are equally important for a successful machine learning model! ## How to share your pre-trained models Please send an email to txie@mit.edu if you want to share your own pre-trained models. Since we don't have time to check the validity of your model, we only accept **peer reviewed** works. To submit, be sure to include: 1. A `pth.tar` file storing the CGCNN model. 2. The type of the model and the target property. 3. The links to the data reference and model reference.

近期下载者

相关文件


收藏者