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

Merge pull request #217 from nerdcha/master

Browse Source 
Bugfix for multiclass sklearn wrapper
This commit is contained in:
Tianqi Chen 2015-04-02 21:14:21 -07:00
commit 93d3f4fe61
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
wrapper/xgboost.py
View File

@ -829,7 +829,7 @@ class XGBClassifier(XGBModel, ClassifierMixin):
def predict_proba(self, X): def predict_proba(self, X):
testDmatrix = DMatrix(X) testDmatrix = DMatrix(X)
class_probs = self._Booster.predict(testDmatrix) class_probs = self._Booster.predict(testDmatrix)
if self._yspace == "multiclass": if self.objective == "multi:softprob":
return class_probs return class_probs
else: else:
classone_probs = class_probs classone_probs = class_probs