Chemprop

Chemprop implements directed message-passing neural networks (D-MPNNs) to predict chemical and molecular properties from structural representations for molecular property prediction and computational chemistry analyses.


Key Features:

  • Directed message-passing neural networks (D-MPNNs): Implements D-MPNNs for graph-based deep learning on molecular structures.
  • Supported data types: Handles multimolecule properties, reactions, atom-level properties, and spectra.
  • Uncertainty quantification and calibration: Includes uncertainty quantification and calibration methods to assess prediction reliability.
  • Pretraining and transfer learning: Supports pretraining and transfer learning workflows to leverage existing data and models.
  • Hyperparameter optimization: Provides improved hyperparameter optimization techniques for model tuning.
  • Customizable model components: Allows customization of loss functions and atom/bond feature sets.
  • Benchmarking: Benchmarked on property prediction datasets including MoleculeNet and SAMPL.

Scientific Applications:

  • Molecular property prediction: Predicts properties such as water-octanol partition coefficients.
  • Reaction property prediction: Predicts reaction barrier heights.
  • Atom-level property prediction: Predicts atomic partial charges.
  • Spectral prediction: Predicts absorption spectra.
  • Benchmarking and validation: Used for benchmarking on datasets including MoleculeNet and SAMPL.

Methodology:

Implements directed message-passing neural networks (D-MPNNs) and incorporates uncertainty quantification and calibration methods, pretraining and transfer learning workflows, hyperparameter optimization, and customization of loss functions and atom/bond features.

Topics

Details

License:
MIT
Cost:
Free of charge
Tool Type:
command-line tool
Operating Systems:
Linux
Programming Languages:
Python, Shell, D
Added:
4/19/2024
Last Updated:
11/24/2024

Operations

Publications

Heid E, Greenman KP, Chung Y, Li S, Graff DE, Vermeire FH, Wu H, Green WH, McGill CJ. Chemprop: A Machine Learning Package for Chemical Property Prediction. Journal of Chemical Information and Modeling. 2023;64(1):9-17. doi:10.1021/acs.jcim.3c01250. PMID:38147829. PMCID:PMC10777403.

PMID: 38147829
Funding: - Defense Sciences Office, DARPA: HR00111920025 - KU Leuven: STG/22/032 - Austrian Science Fund: J-4415 - Division of Graduate Education: 1745302

Links