Package: auglichem
AugLiChemWelcome to AugLiChem!
The augmentation library for chemical systems.
This package supports augmentation for both crystaline and molecular systems, as well as provides automatic downloading for our benchmark datasets, and easy to use model implementations.
In depth documentation about how to use AugLiChem, make use of transformations, and train models is given on ourwebsite.InstallationAugLiChem is apython3.8+package.LinuxIt is recommended to use an environment manager such as conda to install AugLiChem.
Instructions can be foundhere.
If using conda, creating a new environment is ideal and can be done simply by running the following command:conda create -n auglichem python=3.8Then activating the new environment withconda activate auglichemAugLiChem is built primarily withpytorchand that should be installed independently according to your system specifications.
After activating your conda environment,pytorchcan be installed easily and instructions are foundhere.torch_geometricneeds to be installed withconda install pyg -c pyg -c conda-forge.Once you havepytorchandtorch_geometricinstalled, installing AugLiChem can be done using PyPI:pip install auglichemMacOS ARM64 ArchitectureA more involved install is required to run on the new M1 chips since some of the packages do not     have official support yet.
We are working on a more elegant solution given the current limitations.First, download this repo.If you do not have it yet,, conda for ARM64 architecture needs to be installed.
This can be done with Miniforge (which contains conda installer) which is installed by following     the guidehereOnce you have miniforge compatible with ARM64 architecture, a new environment with rdkit can be i    nstalled.
If you do not specifypython=3.8it will default topython=3.9.6as of the time of writing th    is.conda create -n auglichem python=3.8 rdkitNow activate the environment:conda activate auglichemFrom here, individual packages can be installed:conda install -c pytorch pytorchconda install -c fastchan torchvisionconda install scipyconda install cythonconda install scikit-learnpip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cpu.htmlpip install torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cpu.htmlpip install torch-geometricBefore installing the package, you must go intosetup.pyin the main directory and comment outrdkit-pypiandtensorboardfrom theinstall_requireslist since they are already installed.
Not commenting these packages out will result in an error during installation.Finally, run:pip install .Usage guides are provided in theexamples/directory and provide useful guides for using both the molecular and crystal sides of the package.
Make sure to installjupyterbefore working with examples, usingconda install jupyter.
After installing the package as described above, the example notebooks can be downloaded separately and run locally.AuthorsRishikesh Magar*, Yuyang Wang*, Cooper Lorsung*, Hariharan Ramasubramanian, Chen Liang, Peiyuan Li, Amir Barati Farimani*Equal contributionPaperOur paper can be foundhereCitationIf you use AugLiChem in your work, please cite:@article{Magar_2022,
doi = {10.1088/2632-2153/ac9c84},
url = {https://dx.doi.org/10.1088/2632-2153/ac9c84},
year = {2022},
month = {nov},
publisher = {IOP Publishing},
volume = {3},
number = {4},
pages = {045015},
author = {Rishikesh Magar and Yuyang Wang and Cooper Lorsung and Chen Liang and Hariharan Ramasubramanian and Peiyuan Li and Amir Barati Farimani},
title = {AugLiChem: data augmentation library of chemical structures for machine learning},
journal = {Machine Learning: Science and Technology},
abstract = {Machine learning (ML) has demonstrated the promise for accurate and efficient property prediction of molecules and crystalline materials. To develop highly accurate ML models for chemical structure property prediction, datasets with sufficient samples are required. However, obtaining clean and sufficient data of chemical properties can be expensive and time-consuming, which greatly limits the performance of ML models. Inspired by the success of data augmentations in computer vision and natural language processing, we developed AugLiChem: the data augmentation library for chemical structures. Augmentation methods for both crystalline systems and molecules are introduced, which can be utilized for fingerprint-based ML models and graph neural networks (GNNs). We show that using our augmentation strategies significantly improves the performance of ML models, especially when using GNNs. In addition, the augmentations that we developed can be used as a direct plug-in module during training and have demonstrated the effectiveness when implemented with different GNN models through the AugliChem library. The Python-based package for our implementation of Auglichem: Data augmentation library for chemical structures, is publicly available at: https://github.com/BaratiLab/AugLiChem.},highly accurate ML models for chemical str$
}LicenseAugLiChem is MIT licensed, as found in theLICENSEfile. Please note that some of the dependencies AugLiChem uses may be licensed under different terms.