Skip to content

Instantly share code, notes, and snippets.

@risenW

risenW/modelling_randforest.py

Created July 10, 2019 21:35
Show Gist options
  • Save risenW/f291b9d806a57c790b87742ca83b7f3c to your computer and use it in GitHub Desktop.
Save risenW/f291b9d806a57c790b87742ca83b7f3c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
modelling_randforest.py
from sklearn.ensemble import RandomForestRegressor
rf_model = RandomForestRegressor(n_estimators=100, max_depth=5)
print("MAE Sccore: ", cross_validate(rf_model, 10, X_train, y_train))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment