从GridSearchCV获取功能重要性

时间:2018-01-22 08:23:49

标签: 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)

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

也许像

optimized_GBM.feature_importances_

3 个答案:

答案 0 :(得分:13)

这个有效

optimized_GBM.best_estimator_.feature_importances_

答案 1 :(得分:6)

知道了。它是这样的:

optimized_GBM.best_estimator_.feature_importance()

答案 2 :(得分:1)

这取决于您选择的型号。如果选择SVM,则不会具有功能重要性参数,但是在决策树中会得到它