我正在对300K个样本和19个特征进行二进制分类。我在scikit中使用了RandomizedLogisticRegression()进行功能选择。我想知道如何找到RandomizedLogisticRegression()选择了哪些功能。
您应该使用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]