python - 如何从 Sklearn 管道中提取特征重要性

Question

我在 Scikit-Learn 中构建了一个管道，分为两个步骤:一个是构建特征，第二个是 RandomForestClassifier。

虽然我可以保存该管道，但查看各个步骤和步骤中设置的各种参数，我希望能够从生成的模型中检查特征重要性。

那可能吗？

Answer 1

啊，是的。

您列出了要检查估算器的步骤:

例如:

pipeline.steps[1]

返回:

('predictor',
 RandomForestClassifier(bootstrap=True, class_weight=None, criterion='gini',
             max_depth=None, max_features='auto', max_leaf_nodes=None,
             min_samples_leaf=1, min_samples_split=2,
             min_weight_fraction_leaf=0.0, n_estimators=50, n_jobs=2,
             oob_score=False, random_state=None, verbose=0,
             warm_start=False))

然后您可以直接访问模型步骤:

pipeline.steps[1][1].feature_importances_