Improved supported for `model_metrics` outside of XGB and RF. Should work for all scikit-learn models now
