CatBoost Early Stopping To Avoid Overfitting

Python

 1|  from catboost import CatBoostRegressor
 2|  
 3|  # Step 1: Initialise CatBoost regression model
 4|  model = CatBoostRegressor(loss_function='RMSE',
 5|                            n_estimators=1000,
 6|                            random_seed=101)
 7|  
 8|  # Step 2: Declare evaluation set
 9|  eval_set = (X_test, y_test)
10|  
11|  # Step 3: Fit model with early stopping rounds set to 10
12|  model.fit(X_train,
13|            y_train,
14|            eval_set=eval_set,
15|            early_stopping_rounds=10,
16|            verbose=False)
detro
Catboost | Python | Early stopping
Did you find this snippet useful?

Sign up for free to to add this to your code library

Remove Stop Words from Text in DataFrame Column
Python

Pandas | Nltk | Nlp

133
LightGBM Custom Loss Function
Python

LightGBM | Loss function

121
Tuning XGBoost Hyperparameters with Grid Search
Python

XGBoost | Tuning

117
How to Convert DataFrame Values Into Percentages
Python

Lambda | Percentages | Pipe

109
How to Scale Data Using Standard Scaler But Keep Column Names
Python

Scaler | Standard

79
LightGBM Hyperparameter Tuning with GridSearch
Python

Python | LightGBM | Hyperparameter tuning | Gridsearch

77
Calculating Root Mean Squared Error (RMSE) with Sklearn and Python
Python

Rmse

77
Dynamically Create Columns in Pandas Dataframe
Python

Pandas | Columns

61
How to Train a Catboost Classifier with GridSearch Hyperparameter Tuning
Python

Catboost | Hyperparameter tuning

60
Overwriting a Python Logging File Instead of Appending
Python

Logging

55