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
Please refer to our documentation page for installation instructions and full API documentation. You can also view some example scripts we put together.
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 (Travis_Kessler@student.uml.edu) and/or John Hunter Mack (Hunter_Mack@uml.edu).