Skip to content

Instantly share code, notes, and snippets.

@risenW

risenW/modelling_extratrees.py

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