0
我使用的是从feature selection documentaion开始的基于L1的特征选择。转换后的结果给出了一个numpy数组。有没有一种方法可以找出在转换输出X_new
中选择了哪些功能。找出特征选择中已转换输出中的选定特征
from sklearn.svm import LinearSVC
from sklearn.datasets import load_iris
from sklearn.feature_selection import SelectFromModel
import pandas as pd
iris = load_iris()
y=iris.target
X = pd.DataFrame(iris.data,columns=['sepal_length','sepal_width','petal_length','petal_width'])
print X.shape #(150,4)
lsvc = LinearSVC(C=0.01, penalty="l1", dual=False).fit(X, y)
model = SelectFromModel(lsvc, prefit=True)
X_new = model.transform(X)
print X_new.shape #(150, 3)