Use `Optuna.integration.CatBoostPruningCallback` in `catboost_pruning.py`
nzw0301 opened this issue · 0 comments
nzw0301 commented
Motivation
Thanks to optuna/optuna#2734, Optuna provides the pruner callback of catboost as an integration class.
This PR is based on catboost_pruning.py
in this example repo. Hence we can simplify the example code by using the integration class.
Description
Rewrite catboost_pruning.py
by using
optuna.integration.CatBoostPruningCallback
.