epatyukova's picture
Update README.md
7558828 verified
metadata
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