Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change type of xgbclassifier.classes_ from list to numpy array

Browse Source
This commit is contained in:
DAndrey 2016-03-17 16:54:33 +03:00
parent 612ccd0bb7
commit 311f7c8f47
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python-package/xgboost/sklearn.py
View File

@ -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_)