Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/CrawfordGroup/MLQM

Machine-learning quantum mechanics
https://github.com/CrawfordGroup/MLQM

Last synced: 3 months ago
JSON representation

Machine-learning quantum mechanics

Awesome Lists containing this project

README

        

# MLQM: Machine-learning quantum mechanics

A python module centered around developing, testing, and deploying machine-learning algorithms to quantum chemical datasets. We have a focus on wave function-based descriptors, with capabilities for building representations from ab initio electronic structure data such as wave function amplitudes or reduced density matrices.

Beyond representation generation, some `scikit-learn` functions are wrapped for ease-of-use with the `Dataset` class, such as k-means clustering and kernel ridge regression. Defaults are chosen based on performance; however, additional options are available, and `scikit-learn` may also be used directly. See docstrings for details.

To install, run:
```python
pip install -e .
```

To test, run:
```python
py.test
```
from the base `mlqm/` directory.