Skip to content

Instantly share code, notes, and snippets.

@risenW

risenW/modelling_catboost.py

Last active July 17, 2019 09:10
Show Gist options
  • Save risenW/a8162bc6874bd40d95fb4c6e9092bd02 to your computer and use it in GitHub Desktop.
Save risenW/a8162bc6874bd40d95fb4c6e9092bd02 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
modelling_catboost.py
from catboost import CatBoostRegressor
cat_model = CatBoostRegressor(iterations=200,learning_rate=0.1,depth=4)
print("MAE Sccore: ", cross_validate(cat_model, 10, X_train, y_train))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment