Skip to content

A package for density functional approximation using machine learning.

License

Notifications You must be signed in to change notification settings

zhonger/ml-dft

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ML-DFT: Machine learning for density functional approximations This repository contains the implementation for the kernel ridge regression based density functional approximation method described in the paper "Quantum chemical accuracy from density functional approximations via machine learning".

Requirements:
  • python==3.7
  • numpy>=1.19
  • scikit-learn>=0.23
  • scipy>=1.5
  • sacred==0.8
  • ase>=3.20
Installation:

Clone the repository by running:

git clone https://github.com/MihailBogojeski/ml-dft.git

To install package along with all requirements, simply run:

cd ml-dft
pip install -r requirements.txt
pip install .

Install time should range from 3-10 minutes, depending on internet speed.

Training and testing a basic model

To train and test an example model based on a small dataset consisting of 102 water geometries, use the following command:

python run_ml_dft_experiment.py water_102 water_102 50 50

The training and evalutaion of this example model should be done in under a minute. The final lines of the output should look like:

Energies:
Correlation:  0.9999821536536118
RMSE:  0.15695036938603066
MAE:  0.08933831997099333
Max MAE:  0.6630057379879872

Since the crossvalidation process is random the values can be different, however if the model is trained successfully the MAE should be in the range from 0.2 to 0.06.

About

A package for density functional approximation using machine learning.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%