我正在使用 sklearn 的GridSearchCV来为我的随机森林模型获取最佳参数。
下面是我的代码
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_)
我的结果数据框如下
如果你观察到 mymean_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,)
陪伴而非守候
相关分类