Implementation codes for Boosting-Crystal Graph Convolutional Neural Network (Boosting-CGCNN) Link.
- Make a conda environment
conda create -n gb python=3.9
conda activate gb
- Install dependencies
git clone https://github.com/sunilegna/GB-CGCNN.git
pip install poetry
poetry install
- Training code
python main.py
If you use this code, please reference the following paper:
- Eun Ho Kim, Jun Hyeong Gu, June Ho Lee, Seong Hun Kim, Jaeseon Kim, Hyo Gyeong Shin, Shin Hyun Kim, Donghwa Lee. Boosting–crystal graph convolutional neural network for predicting highly imbalanced data: a case study for metal-insulator transition materials. ACS Appl. Mater. Interfaces, 16(33), 43734-43741 (2024). https://pubs.acs.org/doi/10.1021/acsami.4c07851
