o }oi@s^ddlmZddlZddlmZddlZddlmZddlm Z dd d Z Gd d d eZ dS)) annotationsN)Any) BasePruner)_is_first_in_interval_stepvaluerreturnfloatc Cs>zt|}W|Sttfydt|jd}t|dw)Nz!The `value` argument is of type 'z' but supposed to be a float.)r TypeError ValueErrortype__name__)rmessagerM/home/ubuntu/.local/lib/python3.10/site-packages/optuna/pruners/_threshold.py _check_value s  rc@s.eZdZdZ    ddd dZdddZdS)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. Nrlower float | Noneuppern_warmup_stepsintinterval_stepsrNonecCs|dur |dur td|durt|}|durt|}|dur"|ntd }|dur-|ntd}||kr9td|dkrEtd|d|dkrQtd|d||_||_||_||_dS) Nz(Either lower or upper must be specified.infz#lower should be smaller than upper.rz2Number of warmup steps cannot be negative but got .rz2Pruning interval steps must be at least 1 but got )r rrr _lower_upper_n_warmup_steps_interval_steps)selfrrrrrrr__init__Qs,   zThresholdPruner.__init__study'optuna.study.Study'trial'optuna.trial.FrozenTrial'boolcCsv|j}|dur dS|j}||krdSt||j||jsdS|j|}t|r+dS||jkr2dS||j kr9dSdS)NFT) last_steprrintermediate_valueskeysrmathisnanrr)r r"r$stepr latest_valuerrrpruners$    zThresholdPruner.prune)NNrr) rrrrrrrrrr)r"r#r$r%rr&)r __module__ __qualname____doc__r!r.rrrrrs: !r)rrrr) __future__rr*typingroptunaoptuna.prunersroptuna.pruners._percentilerrrrrrrs