epatyukova's picture
Update README.md
7558828 verified
---
license: cc-by-4.0
---
These are Quantile Random Forest Regression models trained to predict the kpoints-density with different confidence levels:
1. QRF95.pkl predicts (0.05, 0.5, 0.95) quantiles
2. QRF90.pkl predicts (0.1, 0.5, 0.9) quantiles
3. QRF85.pkl predicts (0.15, 0.5, 0.85) quantiles
The performance of models measured for the 0.5 quantile is:
MAE: 0.064,
MAPE: 0.179,
MSE: 0.0098,
R2_score: 0.694,
Spearman_corr: 0.862,
Kendall_corr: 0.682
Models are trained on the dataset generated for this work.
Associated GitHub repositories:
https://github.com/stfc/goldilocks
https://github.com/stfc/goldilocks_kpoints