supporting code for feature importance study in https://dx.doi.org/10.1063/5.0050885 - View it on GitHub