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

Merge pull request #1003 from DrAndrey/master

Browse Source 
change type of xgbclassifier.classes_ from list to numpy array
This commit is contained in:
Tianqi Chen 2016-03-17 09:51:05 -07:00
commit c449dc6874
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_)