Skip to content

Instantly share code, notes, and snippets.

@risenW

risenW/modelling_lineardt.py

Created July 10, 2019 21:21
Show Gist options
  • Save risenW/69a7ae439dbf1232dbf15b1302a6a532 to your computer and use it in GitHub Desktop.
Save risenW/69a7ae439dbf1232dbf15b1302a6a532 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
modelling_lineardt.py
from sklearn.tree import DecisionTreeRegressor
dt_model = DecisionTreeRegressor(max_depth=6, min_samples_leaf=2)
print("MAE Sccore: ", cross_validate(dt_model, 10, X_train, y_train))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment