Grid search python

from sklearn.model_selection import GridSearchCV

params = {
    'n_estimators': [5, 10, 20],
    'max_depth': [1, 5, 10],
}

grid_search = GridSearchCV(RandomForestRegressor(), param_grid=params, n_jobs=-1)
grid_search.fit(X_train, y_train)
score = np.round(grid_search.score(X_test, y_test), 3)
print(f"Best random forest classifier score: {score}")
grid_search.best_estimator_, grid_search.best_params_

Similar codes

5 votes191 views1 answer
Pandas save file to pickle
Python
0 vote227 views1 answer
How to use tensorboard
Python
0 vote221 views1 answer
Print map object python
Python
1 vote182 views1 answer
Creating dictionary using the keys
Python
1 vote135 views1 answer
Python ignore return value
Python
16 votes189 views2 answers
Choice random python
Python
2 votes190 views1 answer
How to make file to curent dir python
Python
1 vote190 views1 answer
Swapping of two numbers in python
Python
1 vote217 views2 answers
Keras sequential layer
Python
12 votes204 views2 answers
How to read input from stdin in python
Python