XGBoost wrapper added in classification_utils