added test to expose the predict_proba w/ sparse matrix regression

pprett · GaelVaroquaux · commit b7f6f194a329 · 2012-10-06T15:47:16.000+02:00
diff --git a/sklearn/linear_model/tests/test_sgd.py b/sklearn/linear_model/tests/test_sgd.py
@@ -28,6 +28,10 @@ def decision_function(self, X, *args, **kw):
         X = sp.csr_matrix(X)
         return SGDClassifier.decision_function(self, X, *args, **kw)
 
+    def predict_proba(self, X, *args, **kw):
+        X = sp.csr_matrix(X)
+        return SGDClassifier.predict_proba(self, X, *args, **kw)
+
 
 class SparseSGDRegressor(SGDRegressor):
 
