Disable feature validation on sklearn predict prob. (#5953)

* Fix issue when scikit learn interface receives transformed inputs.
2020-07-29 19:26:44 +08:00
parent 18349a7ccf
commit f5fdcbe194
2 changed files with 33 additions and 1 deletions
--- a/python-package/xgboost/sklearn.py
+++ b/python-package/xgboost/sklearn.py
@@ -909,7 +909,7 @@ class XGBClassifier(XGBModel, XGBClassifierBase):
            'Label encoder is not defined.  Returning class probability.')
        return class_probs

-    def predict_proba(self, data, ntree_limit=None, validate_features=True,
+    def predict_proba(self, data, ntree_limit=None, validate_features=False,
                      base_margin=None):
        """
        Predict the probability of each `data` example being of a given class.