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

bugfix

Browse Source
This commit is contained in:
Jamie Hall 2015-04-02 20:33:07 -07:00
parent 9b0dee986f
commit d17cdd639f
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):
testDmatrix = DMatrix(X)
class_probs = self._Booster.predict(testDmatrix)
if self._yspace == "multiclass":
if self.objective == "multi:softprob":
return class_probs
else:
classone_probs = class_probs