We need to make the same changes to `explain_feature_prediction` that we did for the other functions to add compatibility to XGBoost