通过RandomizedLogisticRegression查找选定的功能



我正在对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]

相关内容

  • 没有找到相关文章

最新更新