/RDF-KRR-DOS

Predicting DOS at Fermi level with RDF as descriptor and KRR as the ML algorithm.

Primary LanguageJupyter Notebook

RDF ---> Density of States (DOS) at Fermi level.

A machine learning project to predict DOS at fermi level of sp and spd systems using RDF as descriptor. This project is inspired by: https://journals.aps.org/prb/abstract/10.1103/PhysRevB.89.205118

Descriptor: Radial distribution function (RDF)

Prediction: DOS at Fermi level

1. Create RDF code

A RDF code (RDF.py) is created and tested with XCl (X = Li, Na, K) compounds. Check out the result here: https://github.com/yanxon/RDF/blob/master/XCl_test.ipynb

2. Machine learning procedure

https://github.com/yanxon/RDF-DOS-KRR/blob/master/RDF_DOS_KRR.py (RDF ---> DOS, Aflow)

This script extracts sp metals from AFLOW database. The sp metals are split into train and test datasets. Finally, KRR algorithm is used to train the train dataset and tested with the test dataset.

3. Result

sp metals:

krr_0 1

cap: MAE = 0.0022 states/eV/A^3; r^2 = 0.6540

Comparison to PRB (spd system):

krr_prb_mp

cap: MAE = 0.01258 states/eV/A^3; r^2 = 0.6748