def __call__(self, study: optuna.study.Study, trial: optuna.trial.FrozenTrial) -> None:
if trial.state == optuna.trial.TrialState.PRUNED:
self._consequtive_pruned_count += 1
else:
self._consequtive_pruned_count = 0
if self._consequtive_pruned_count >= self.threshold:
study.stop()
def callback(study: optuna.study.Study,
trial: optuna.trial.FrozenTrial) -> None:
if trial.number >= 4:
study.stop()
