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

Merge pull request #319 from jdwittenauer/master

Browse Source 
Add classes_ attribute to scikit-learn wrapper
This commit is contained in:
Tianqi Chen 2015-05-15 22:03:18 -07:00
commit f49525ee95
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wrapper/xgboost.py
View File

@ -935,8 +935,8 @@ class XGBClassifier(XGBModel, XGBClassifier):
base_score, seed)
def fit(self, X, y, sample_weight=None):
y_values = list(np.unique(y))
self.n_classes_ = len(y_values)
self.classes_ = list(np.unique(y))
self.n_classes_ = len(self.classes_)
if self.n_classes_ > 2:
# Switch to using a multiclass objective in the underlying XGB instance
self.objective = "multi:softprob"