python - CatBoost出现过拟合后，有办法保存训练好的模型吗？

Question

我正在使用 CatBoostRegressor在 Python 版本的 Catboost 库中。

根据文档，可以使用我正在做的过拟合检测器，如下所示:

model = CatBoostRegressor(iterations=iters, learning_rate=0.03, depth=depth, verbose=True, od_pval=1, od_type='IncToDec', od_wait=20)
model.fit(train_pool, eval_set=validation_pool)

# this code didn't executed
model.save_model(model_name)

然而，过拟合发生后，我的 Python 脚本被中断，过早停止，选择任何你想要的短语，并没有执行保存模型部分，这导致浪费了大量时间并且没有结果结束。我没有得到任何堆栈跟踪信息。

是否有可能在 CatBoost 中处理它并节省数小时的装配工作？

Answer 1

使用此代码。无论 try block 中发生什么，它都会保存模型。

try:
    model.fit(X, y)
finally:
    model.save_model()