有没有办法从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_
答案 0 :(得分:13)
这个有效
optimized_GBM.best_estimator_.feature_importances_
答案 1 :(得分:6)
知道了。它是这样的:
optimized_GBM.best_estimator_.feature_importance()
答案 2 :(得分:1)
这取决于您选择的型号。如果选择SVM,则不会具有功能重要性参数,但是在决策树中会得到它