An interface between molecules and machine learning
A library to interface molecules and machine learning. The goal of this library is to be a simple way to convert molecules into a vector representation for later use with libraries such as scikit-learn. This is done using a similar API scheme.
All of the coordinates are assumed to be in angstroms.
- Simple interface to many common molecular descriptors and their variants - Molecule - Coulomb Matrix - Bag of Bonds - Encoded Bonds - Encoded Angles - Connectivity Counts - Atom - Shell - Local Encoded Bonds - Local Encoded Angles - Local Coulomb Matrix - Behler-Parrinello - Kernel - Atom Kernel - Crystal - Generallized Crystal - Ewald Sum Matrix - Sine Matrix - Parallel feature generation - Ability to save/load fit models - Multiple input formats supported (and ability to define your own) - Supports both Python 2 and Python 3
>>> from molml.features import CoulombMatrix >>> feat = CoulombMatrix() >>> H2 = ( ... ['H', 'H'], ... [ ... [0.0, 0.0, 0.0], ... [1.0, 0.0, 0.0], ... ] ... ) >>> HCN = ( ... ['H', 'C', 'N'], ... [ ... [-1.0, 0.0, 0.0], ... [ 0.0, 0.0, 0.0], ... [ 1.0, 0.0, 0.0], ... ] ... ) >>> feat.fit([H2, HCN]) CoulombMatrix(input_type='list', n_jobs=1) >>> feat.transform([H2]) array([[ 0.5, 1. , 0. , 1. , 0.5, 0. , 0. , 0. , 0. ]]) >>> feat.transform([H2, HCN]) array([[ 0.5 , 1. , 0. , 1. , 0.5 , 0. , 0. , 0. , 0. ], [ 0.5 , 6. , 3.5 , 6. , 36.8581052, 42. , 3.5 , 42. , 53.3587074]])
For more examples, look in the examples. Note: To run some of the examples scikit-learn>=0.16.0 is required.
For the full documentation, refer to the docs or the docstrings in the code.
MolML works with both Python 2 and Python 3. It has been tested with the versions listed below, but newer versions should work.
python>=2.7/3.4/3.5/3.5/3.6 numpy>=1.9.1 scipy>=0.15.1 pathos>=0.2.0 future # For python 2
NOTE: Due to an issue with multiprocess (a pathos dependency), the minimum version of Python that will work is 2.7.4. For full details see this link. Without this, the parallel computation of features will fail.
Once numpy and scipy are installed, the package can be installed with pip.
$ pip install molml
Or for the bleeding edge version, you can use
$ pip install git+git://github.com/crcollins/molml
To install a development version, just clone the git repo.
$ git clone https://github.com/crcollins/molml $ # cd to molml and setup some virtualenv $ pip install -r requirements-dev.txt
To build the documentation, you just need to install the documentation dependencies. These are already included in the dev install.
$ cd docs/ $ pip install -r requirements-docs.txt $ make html
To run the tests, make sure that nose is installed and then run:
To include coverage information, make sure that coverage is installed and then run:
$ nosetests --with-coverage --cover-package=molml --cover-erase