Atom Modeling PipeLine

Abstract

This repository contains code for the computational framework to build ML models for drug discovery purposes. Our end-to-end pipeline extracts features from model-ready datasets and trains and saves the model to our model zoo or to disk. Our pipeline generates a variety of molecular features and both shallow and deep ML models. The HPC-specific module we have developed conducts efficient parallelized search of the model hyperparameter space and reports the best-performing hyperparameters for each of these feature/model combinations.