Я использую sklearn's 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_)
Myданные в результатах приведены ниже
Если вы заметите, мой 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,)