【发布时间】:2019-06-25 00:08:52
【问题描述】:
我正在使用 sklearn 的 GridSearchCV 为我的 Random Forest 模型获取最佳参数。
下面是我的代码
model = RandomForestRegressor(random_state = 1, n_jobs = -1)
param_grid = {"n_estimators": [5, 10]}
for parameter, param_range in dict.items(param_grid):
#get_optimum_range(parameter, param_range, RFReg, index)
grid_search = GridSearchCV(estimator=model, param_grid = {parameter: param_range})
grid_search.fit(X_train, y_train)
results = pd.DataFrame(grid_search.cv_results_)
我的结果数据框如下
如果你观察到我的mean_test_score 是负面的,但mean_train_score 是正面的。
同样的原因可能是什么?
我的数据框大小
print(X_train.shape)
print(y_train.shape)
print(X_test.shape)
print(y_test.shape)
(538, 3)
(538,)
(112, 3)
(112,)
【问题讨论】:
标签: python scikit-learn random-forest grid-search