web-dev-qa-db-ja.com

GridSearchCVから機能の重要性を取得する

SklearnのGridSearchCVから機能の重要性を取得する方法はありますか？

例えば ​​：

from sklearn.model_selection import GridSearchCV
print("starting grid search ......")
optimized_GBM = GridSearchCV(LGBMRegressor(),
                             params,
                             cv=3,
                             n_jobs=-1)
# 
optimized_GBM.fit(tr, yvar)
preds2 = optimized_GBM.predict(te)

機能の重要性にアクセスする方法はありますか？

たぶん

optimized_GBM.feature_importances_
pythonscikit-learngrid-search
8
Nick M

とった。それはこのようなものになります：

optimized_GBM.best_estimator_.feature_importance()

パイプラインを介してこれを実行し、object has no attribute 'feature_importance'を受け取った場合は、optimized_GBM.best_estimator_.named_steps ["step_name"]。feature_importances_を試してください。

ここで、step_nameはパイプライン内の対応する名前です

7
Nick M

これはうまくいく

optimized_GBM.best_estimator_.feature_importances_
15
Jaqen H'ghar