diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py index 910fd8886..2e8157ff6 100644 --- a/python-package/xgboost/sklearn.py +++ b/python-package/xgboost/sklearn.py @@ -363,7 +363,7 @@ class XGBClassifier(XGBModel, XGBClassifierBase): metric measured on the validation set to stderr. """ evals_result = {} - self.classes_ = list(np.unique(y)) + self.classes_ = np.unique(y) self.n_classes_ = len(self.classes_)