From 117faf6590d13ea0bc434d802b650e91c16ee937 Mon Sep 17 00:00:00 2001
From: Alexandre Gramfort <alexandre.gramfort@inria.fr>
Date: Wed, 28 Apr 2010 07:12:39 +0000
Subject: [PATCH] adding predict_proba method in UnivSelection class
git-svn-id: https://scikit-learn.svn.sourceforge.net/svnroot/scikit-learn/trunk@709 22fbfee3-77ab-4535-9bad-27d1bd3bc7d8
---
scikits/learn/feature_selection/univ_selection.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/scikits/learn/feature_selection/univ_selection.py b/scikits/learn/feature_selection/univ_selection.py
index 3272c779a0..ee1f59a70b 100644
--- a/scikits/learn/feature_selection/univ_selection.py
+++ b/scikits/learn/feature_selection/univ_selection.py
@@ -291,6 +291,11 @@ class UnivSelection(object):
else:
return self.estimator.predict(x[:,support_])
+ def predict_proba(self, X):
+ self.support_ = self.select_func(self.p_values_, *self.select_args)
+ support_ = self.support_
+ return self.estimator.predict_proba(X[:,support_])
+
class UnivSelect(object):
def __init__(self, score_func=f_regression,
--
GitLab