通过RandomizedLogisticRegression查找所选要素

时间:2015-06-26 17:29:19

标签: python scikit-learn logistic-regression feature-selection

我正在对300Ksamples和19个功能进行二进制分类。 我在scikit中使用RandomizedLogisticRegression()进行特征选择。 我想知道如何找到RandomizedLogisticRegression()选择的功能。

1 个答案:

答案 0 :(得分:4)

您应该使用get_support功能:

from sklearn.datasets import load_iris
from sklearn.linear_model import RandomizedLogisticRegression

iris = load_iris()
X, y = iris.data, iris.target

clf = RandomizedLogisticRegression()
clf.fit(X,y)
print clf.get_support()

#prints [False  True  True  True]

或者,您可以获取支持功能的索引:

print clf.get_support(indices=True)
#prints [1 2 3]