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

16 votes1.1K views4 answers
Tqdm library
Python
2 votes173 views2 answers
Cv2 crop image
Python
4 votes73 views1 answer
Install unittest python
Python
0 vote113 views1 answer
How to find how many processors you have with python
Python
8 votes127 views1 answer
Negative lookbehind javascript
Python
0 vote119 views1 answer
Python pandas get dataframe element
Python
0 vote78 views1 answer
FunÃ§Ã£o anonima python
Python
0 vote111 views1 answer
Split dataset into train, test and validation sets
Python
0 vote128 views1 answer
Print numpy version
Python
1 vote133 views2 answers
Python return using if
Python