o }oi@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddl mZerJdd lmZdd lmZdd lmZdd l mZerZdd l mZddl mZeeZdZdZedddde fd&d d!Zd'd$d%ZdS)() annotations) TYPE_CHECKING)experimental_func) get_logger)DEFAULT_MIN_N_TRIALS)_get_improvement_info) _get_y_range)_imports)Study)BaseErrorEvaluator)BaseImprovementEvaluator)_ImprovementInfo)Axes)pltg?g?z3.2.0FNstudyr plot_errorboolimprovement_evaluatorBaseImprovementEvaluator | Noneerror_evaluatorBaseErrorEvaluator | None min_n_trialsintreturn'Axes'cCs tt||||}t||S)aPlot the potentials for future objective improvement. This function visualizes the objective improvement potentials, evaluated with ``improvement_evaluator``. It helps to determine whether we should continue the optimization or not. You can also plot the error evaluated with ``error_evaluator`` if the ``plot_error`` argument is set to :obj:`True`. Note that this function may take some time to compute the improvement potentials. .. seealso:: Please refer to :func:`optuna.visualization.plot_terminator_improvement`. Args: study: A :class:`~optuna.study.Study` object whose trials are plotted for their improvement. plot_error: A flag to show the error. If it is set to :obj:`True`, errors evaluated by ``error_evaluator`` are also plotted as line graph. Defaults to :obj:`False`. improvement_evaluator: An object that evaluates the improvement of the objective function. Default to :class:`~optuna.terminator.RegretBoundEvaluator`. error_evaluator: An object that evaluates the error inherent in the objective function. Default to :class:`~optuna.terminator.CrossValidationErrorEvaluator`. min_n_trials: The minimum number of trials before termination is considered. Terminator improvements for trials below this value are shown in a lighter color. Defaults to ``20``. Returns: A :class:`matplotlib.axes.Axes` object. )r checkr_get_improvement_plot)rrrrrinfork/home/ubuntu/.local/lib/python3.10/site-packages/optuna/visualization/matplotlib/_terminator_improvement.pyplot_terminator_improvements* r rr cCst|j}tjdt\}}|d|d|dt d}|dkr0t d|S|j |jd|d|j d|dd |dt||krMdndd ||krk|j |j|d|j |dd |ddd |jdur~|j |j|jd |d d d ||t|||S)NggplotzTerminator Improvement PlotTrialzTerminator Improvementtab10rzThere are no complete trials.o)markercoloralphalabel)r&r'r)Error)len trial_numbersrstyleusesubplots set_title set_xlabel set_ylabelget_cmap_loggerwarningplot improvementsALPHAerrorslegendset_ylimr)rrn_trials_axcmaprrrrOsJ            r) rr rrrrrrrrrr)rr rrrr) __future__rtypingroptuna._experimentalroptuna.loggingr'optuna.terminator.improvement.evaluatorr,optuna.visualization._terminator_improvementrr3optuna.visualization.matplotlib._matplotlib_importsr optuna.study.studyr optuna.terminatorr r r is_successfulrr__name__r5PADDING_RATIO_Yr9r rrrrrs4             /