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.best_estimator_.feature_importance()
Get feature importance from GridSearchCV
最新推荐文章于 2023-08-06 23:32:49 发布