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

Merge branch 'master' of ssh://github.com/dmlc/xgboost

Browse Source
This commit is contained in:
tqchen 2015-05-15 23:54:40 -07:00
commit 39f1da08d2
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) base_score, seed)
def fit(self, X, y, sample_weight=None): def fit(self, X, y, sample_weight=None):
y_values = list(np.unique(y)) self.classes_ = list(np.unique(y))
self.n_classes_ = len(y_values) self.n_classes_ = len(self.classes_)
if self.n_classes_ > 2: if self.n_classes_ > 2:
# Switch to using a multiclass objective in the underlying XGB instance # Switch to using a multiclass objective in the underlying XGB instance
self.objective = "multi:softprob" self.objective = "multi:softprob"