:jddlmZddlZddlmZddlmZddlmZddlm Z er ddl m Z ddl m Z d d ZGd d eZy) ) annotationsN)Any) TYPE_CHECKING) BasePruner)_is_first_in_interval_step)Study) FrozenTrialc t|}|S#ttf$r&dt|jd}t|dwxYw)Nz!The `value` argument is of type 'z' but supposed to be a float.)float TypeError ValueErrortype__name__)valuemessages n/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/optuna/pruners/_threshold.py _check_valuersY+e  L z "+/U 0D0D/EEb c  d* +s 5Ac<eZdZdZ d ddZddZy)ThresholdPruneraPruner to detect outlying metrics of the trials. Prune if a metric exceeds upper threshold, falls behind lower threshold or reaches ``nan``. Example: .. testcode:: from optuna import create_study from optuna.pruners import ThresholdPruner from optuna import TrialPruned def objective_for_upper(trial): for step, y in enumerate(ys_for_upper): trial.report(y, step) if trial.should_prune(): raise TrialPruned() return ys_for_upper[-1] def objective_for_lower(trial): for step, y in enumerate(ys_for_lower): trial.report(y, step) if trial.should_prune(): raise TrialPruned() return ys_for_lower[-1] ys_for_upper = [0.0, 0.1, 0.2, 0.5, 1.2] ys_for_lower = [100.0, 90.0, 0.1, 0.0, -1] study = create_study(pruner=ThresholdPruner(upper=1.0)) study.optimize(objective_for_upper, n_trials=10) study = create_study(pruner=ThresholdPruner(lower=0.0)) study.optimize(objective_for_lower, n_trials=10) Args: lower: A minimum value which determines whether pruner prunes or not. If an intermediate value is smaller than lower, it prunes. upper: A maximum value which determines whether pruner prunes or not. If an intermediate value is larger than upper, it prunes. n_warmup_steps: Pruning is disabled if the step is less than the given number of warmup steps. interval_steps: Interval in number of steps between the pruning checks, offset by the warmup steps. If no value has been reported at the time of a pruning check, that particular check will be postponed until a value is reported. Value must be at least 1. Nc<| | td| t|}| t|}||n td }||n td}||kDr td|dkrtd|d|dkrtd|d||_||_||_||_y) Nz(Either lower or upper must be specified.infz#lower should be smaller than upper.rz2Number of warmup steps cannot be negative but got .z2Pruning interval steps must be at least 1 but got )r rr r _lower_upper_n_warmup_steps_interval_steps)selfloweruppern_warmup_stepsinterval_stepss r__init__zThresholdPruner.__init__Vs =U]FG G   'E   'E*u *e  5=BC C A D^DTTUV  A D^DTTUV   --c2|j}|y|j}||kryt||jj ||j sy|j|}t j|ry||jkry||jkDryy)NFT) last_steprrintermediate_valueskeysrmathisnanrr)rstudytrialstepr! latest_values rprunezThresholdPruner.prunews <-- . ) %++002NDDXDX 006 ::l # $++ % $++ %r$)NNrr) r float | Noner r0r!intr"r1returnNone)r+rr,r r2bool)r __module__ __qualname____doc__r#r/r$rrrsQ6t#" ... .  .  .Br$r)rrr2r ) __future__rr)typingrroptuna.prunersroptuna.pruners._percentiler optuna.studyr optuna.trialr rrr8r$rr?s5"  %A"( rjrr$