From 311f7c8f47ec621243db6e68a95589afd78ba360 Mon Sep 17 00:00:00 2001 From: DAndrey Date: Thu, 17 Mar 2016 16:54:33 +0300 Subject: [PATCH] change type of xgbclassifier.classes_ from list to numpy array --- python-package/xgboost/sklearn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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_)