从GridSearchCV获取功能重要性

Nic*_*k M 8 python scikit-learn grid-search

有没有办法从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)
Run Code Online (Sandbox Code Playgroud)

有没有办法可以访问功能重要性?

也许是这样的

optimized_GBM.feature_importances_
Run Code Online (Sandbox Code Playgroud)

Jaq*_*har 15

这个工作

optimized_GBM.best_estimator_.feature_importances_
Run Code Online (Sandbox Code Playgroud)

  • 另外:如果您使用的是管道,即您的估计器是管道对象,则必须添加管道步骤名称:`optimized_GBM.best_estimator_.named_steps["step_name"].feature_importances_` (8认同)
  • @aptha-gowda 有没有办法也提取特征名称?即变量的名称? (2认同)

Nic*_*k M 6

得到它了.它是这样的:

optimized_GBM.best_estimator_.feature_importance()
Run Code Online (Sandbox Code Playgroud)

  • 您是否可能得到“对象没有属性'feature_importance'”错误? (4认同)