optuna.terminator.Terminator

class optuna.terminator.Terminator(improvement_evaluator=None, error_evaluator=None, min_n_trials=20)

Optuna 学习（study）的自动停止机制。

此类实现了 Optuna 学习（study）的自动停止机制，旨在防止不必要的计算。当统计误差（例如交叉验证误差）超出优化的剩余空间时，学习将被终止。

有关该算法的更多信息，请参考以下论文

• A. Makarova 等人。超参数优化的自动终止。

参数：

• improvement_evaluator (BaseImprovementEvaluator | None) – 用于评估优化剩余空间的评估器对象。默认为 RegretBoundEvaluator 对象。

• error_evaluator (BaseErrorEvaluator | None) – 用于计算统计误差（例如交叉验证误差）的评估器。默认为 CrossValidationErrorEvaluator 对象。

• min_n_trials (int) – 在考虑终止之前所需的最小试验次数。默认为 20。

引发：

ValueError – 如果 min_n_trials 不是正整数。

示例

import logging
import sys

from sklearn.datasets import load_wine
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import cross_val_score
from sklearn.model_selection import KFold

import optuna
from optuna.terminator import Terminator
from optuna.terminator import report_cross_validation_scores


study = optuna.create_study(direction="maximize")
terminator = Terminator()
min_n_trials = 20

while True:
    trial = study.ask()

    X, y = load_wine(return_X_y=True)

    clf = RandomForestClassifier(
        max_depth=trial.suggest_int("max_depth", 2, 32),
        min_samples_split=trial.suggest_float("min_samples_split", 0, 1),
        criterion=trial.suggest_categorical("criterion", ("gini", "entropy")),
    )

    scores = cross_val_score(clf, X, y, cv=KFold(n_splits=5, shuffle=True))
    report_cross_validation_scores(trial, scores)

    value = scores.mean()
    logging.info(f"Trial #{trial.number} finished with value {value}.")
    study.tell(trial, value)

    if trial.number > min_n_trials and terminator.should_terminate(study):
        logging.info("Terminated by Optuna Terminator!")
        break

另请参阅

请参阅 TerminatorCallback，了解如何将终止器机制与 optimize() 方法一起使用。

注意

在 v3.2.0 中作为实验性功能添加。接口在更新版本中可能会更改，恕不另行通知。请参阅 https://github.com/optuna/optuna/releases/tag/v3.2.0。

方法

should_terminate(study)

根据报告的值判断是否应该终止学习（study）。

should_terminate(study)

根据报告的值判断是否应该终止学习（study）。

参数：

study (Study)

返回类型：

bool