gpt4 book ai didi

python - RandomForestRegressor 和 feature_importances_ 错误

转载 作者:太空狗 更新时间:2023-10-30 00:53:26 24 4
gpt4 key购买 nike

我正在努力从我的 RandomForestRegressor 中提取特征重要性,我得到:

AttributeError: 'GridSearchCV' object has no attribute 'feature_importances_'.

谁知道为什么没有属性?根据文档应该存在这个属性?

完整代码:

from sklearn.ensemble import RandomForestRegressor
from sklearn.model_selection import GridSearchCV

#Running a RandomForestRegressor GridSearchCV to tune the model.
parameter_candidates = {
'n_estimators' : [650, 700, 750, 800],
'min_samples_leaf' : [1, 2, 3],
'max_depth' : [10, 11, 12],
'min_samples_split' : [2, 3, 4, 5, 6]
}

RFR_regr = RandomForestRegressor()
CV_RFR_regr = GridSearchCV(estimator=RFR_regr, param_grid=parameter_candidates, n_jobs=5, verbose=2)
CV_RFR_regr.fit(X_train, y_train)

#Predict with testing set
y_pred = CV_RFR_regr.predict(X_test)

#Extract feature importances
importances = CV_RFR_regr.feature_importances_

最佳答案

您正在尝试使用 GridSearchCV 对象上的属性。它不在那里。您实际需要做的是访问完成网格搜索的估计器。

通过以下方式访问属性:

importances = CV_RFR_regr.best_estimator_.feature_importances_

关于python - RandomForestRegressor 和 feature_importances_ 错误,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47111434/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com