Skip to content
/ ecnet Public

QSPR-based machine learning for fuel property prediction

License

Notifications You must be signed in to change notification settings

ecrl/ecnet

Repository files navigation

UML Energy & Combustion Research Laboratory

ECNet: machine learning models for fuel property prediction

GitHub version PyPI version status GitHub license Documentation Status

ECNet is an open source Python package for creating machine learning models to predict fuel properties. ECNet comes bundled with a variety of fuel property datasets, including cetane number, yield sooting index, and research/motor octane number. ECNet was built using the PyTorch library, allowing easy implementation of our models in your existing ML pipelines.

ECNet leverages QSPR descriptors for use as input variables, specifically PaDEL-Descriptor and alvaDesc. Using alvaDesc requires a valid license.

Future plans for ECNet include:

  • Implementating RDKit to train using molecular fingerprints
  • Leveraging additional QSPR-generation software packages (e.g. Mordred)
  • A graphical user interface

Installation and Usage

Please refer to our documentation page for installation instructions and full API documentation. You can also view some example scripts we put together.

Contributing, Reporting Issues, and Other Support:

To contribute to ECNet, make a pull request. Contributions should include tests for new features added, as well as extensive documentation.

To report problems with the software or feature requests, file an issue. When reporting problems, include information such as error messages, your OS/environment and Python version.

For additional support/questions, contact Travis Kessler ([email protected]) and/or John Hunter Mack ([email protected]).