From d17cdd639fe2c4f2a601650a694b83058aff9415 Mon Sep 17 00:00:00 2001
From: Jamie Hall <jamie@kaggle.com>
Date: Thu, 2 Apr 2015 20:33:07 -0700
Subject: [PATCH] bugfix

---
 wrapper/xgboost.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wrapper/xgboost.py b/wrapper/xgboost.py
index 8ce5821f4..4a1e7c895 100644
--- a/wrapper/xgboost.py
+++ b/wrapper/xgboost.py
@@ -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