Advertisement
Guest User

Untitled

a guest
Jun 20th, 2018
74
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.56 KB | None | 0 0
  1. param_grid_rfr={'n_estimators':[130, 100, 70, 50],
  2. 'max_depth':[1, 3, 5, 7],
  3. 'criterion': ['mse', 'mae'],
  4. 'min_samples_split':[2, 5, 7],
  5. 'min_samples_leaf':[3, 5, 1, 7, 2],
  6. 'max_features':[1.0 ,0.3, 0.2, 0.4],
  7. 'bootstrap': [True]
  8. }
  9. rfr = RandomForestRegressor()
  10. grid_rfr = GridSearchCV(estimator=rfr, param_grid=param_grid_rfr, scoring='neg_mean_absolute_error', cv=10, verbose=10)
  11. grid_rfr.fit(X_filtered, Y_scaled)
  12. print(grid_rfr.best_score_)
  13. print(grid_rfr.best_estimator_)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement