如何在Python的sklearn中获取不同管道中的特征名称

我使用以下代码(来源)来连接多个特征提取方法。

from sklearn.pipeline import Pipeline, FeatureUnionfrom sklearn.model_selection import GridSearchCVfrom sklearn.svm import SVCfrom sklearn.datasets import load_irisfrom sklearn.decomposition import PCAfrom sklearn.feature_selection import SelectKBestiris = load_iris()X, y = iris.data, iris.targetpca = PCA(n_components=2)selection = SelectKBest(k=1)# Build estimator from PCA and Univariate selection:combined_features = FeatureUnion([("pca", pca), ("univ_select", selection)])# Use combined features to transform dataset:X_features = combined_features.fit(X, y).transform(X)print("Combined space has", X_features.shape[1], "features")svm = SVC(kernel="linear")# Do grid search over k, n_components and C:pipeline = Pipeline([("features", combined_features), ("svm", svm)])param_grid = dict(features__pca__n_components=[1, 2, 3],                  features__univ_select__k=[1, 2],                  svm__C=[0.1, 1, 10])grid_search = GridSearchCV(pipeline, param_grid=param_grid, cv=5, verbose=10)grid_search.fit(X, y)print(grid_search.best_estimator_)

我想从上述代码中获取所选特征的名称。

为此,我使用了grid_search.best_estimator_.support_。然而,这返回了一个错误,显示:

AttributeError: 'Pipeline' object has no attribute 'support_'

在Python的sklearn中,有没有办法获取上述代码中显示的选定特征名称

如果需要,我很乐意提供更多详情。


回答:

这是我了解best_estimator_使用的最终特征的方法

>>> features = grid_search.best_estimator_.named_steps['features']# number of components chosen from pca>>> pca=features.transformer_list[0][1]>>> pca.n_components3# features chosen by selectKbest>>> select_k_best=features.transformer_list[1][1]>>> select_k_best.get_support()array([False, False,  True, False])

Related Posts

关于k折交叉验证的直观问题

我在使用交叉验证检查预测能力时遇到了一些直观问题,我认…

调整numpy数组大小以使用sklearn的train_test_split函数?

我正在尝试使用sklearn中的test_train_…

如何转换二维张量和索引张量以便用于torch.nn.utils.rnn.pack_sequence

我有一组序列,格式如下: sequences = to…

模型预测值的含义是什么?

我在网上找到一个数字识别器的CNN模型并进行了训练,当…

锯齿张量作为LSTM的输入

了解锯齿张量以及如何在TensorFlow中使用它们。…

如何告诉SciKit的LinearRegression模型预测值不能小于零?

我有以下代码,尝试根据非价格基础特征来估值股票。 pr…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注