o }oi,@s0ddlmZddlmZddlmZddlmZddlZddlm Z ddlm Z ddl Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZerfddlmZeeZGdddeZ Gddde Z!Gddde Z"d,d!d"Z#dd#d$d%d-d'd(Z$d.d*d+Z%dS)/) annotations)Callable)Sequence)EnumN)cast) NamedTuple) get_logger)_CONSTRAINTS_KEY)Study)StudyDirection) FrozenTrial) TrialState)_imports)_check_plot_args)goc@seZdZdZdZdZdS) _ValueStaterN)__name__ __module__ __qualname__Feasible Infeasible Incompleterr^/home/ubuntu/.local/lib/python3.10/site-packages/optuna/visualization/_optimization_history.pyrsrc@s.eZdZUded<ded<ded<ded<d S) _ValuesInfoz list[float]valueszlist[float] | Nonestdsstr label_namezlist[_ValueState]statesNrrr__annotations__rrrrr"s  rc@s&eZdZUded<ded<ded<dS)_OptimizationHistoryInfoz list[int] trial_numbersr values_infoz_ValuesInfo | Nonebest_values_infoNr"rrrrr$)s  r$studyStudy | Sequence[Study]target%Callable[[FrozenTrial], float] | None target_namer error_barboolreturnlist[_OptimizationHistoryInfo]c st|||t|tr|g}nt|}g|D]}|}t|dkr%|n|d|j}g}g}|D]K} | jtj krI| t d| t j q3| jt} | dus\tdd| Drc| t jn| t j|duru| || q3| tt | jq3|durd} nBg} |jtjkrddt||D} ttj| } nddt||D} ttj| } t|dkrdnd |j}t| d||} td d|Dt|d||| d qtd krt !d t"ddD}t"ddD}||d kr t !d#|sStd krgS|d kr$t !dgSddD}t$|ddfdd }||d\}}d}|durN|dd\}}t|||gS)Nrz of nancSsg|]}|dkqS)gr).0xrrr Gsz7_get_optimization_history_info_list..cSs&g|]\}}|tjkr |ntdqSinfrrfloatr2vsrrrr4VscSs(g|]\}}|tjkr |ntd qSr5r7r9rrrr4\sz Best ValuezBest Value of cSsg|]}|jqSr)number)r2trrrr4gs)r%r&r'rzThere are no studies.cs |] }|jjtjVqdSN)r&r!countrrr2inforrr p z6_get_optimization_history_info_list..csr>r?)r&r!r@rrrArrrrCsrDzThere are no complete trials.zThere are no feasible trials.cSsg|] }|jD]}|qqSr)r%)r2rBr<rrrr4sr ruse_best_valuer.r/tuple[list[int], _ValuesInfo]csZddtD}ddtD}dusJD]A\}}}|r)|dus'J|}t||j|jD])\}}} t|sS|sJ| tjkrJ|||n |rS|||||| q1qg} g} g} g} tD];}t ||dkrtj||vr| tj| || t || | t || qh| tjqh| t| | || fS)NcSg|]}gqSrrr2_rrrr4zK_get_optimization_history_info_list.._aggregate..cSrGrrrHrrrr4rJr)rangeziprr!mathisinfrrappendlennpmeanitemstdrr)r rErr!r%r&r'nr:r;trial_numbers_union value_states value_means value_stdsi info_list max_num_trialrr _aggregates6      z7_get_optimization_history_info_list.._aggregateFT)r rrEr.r/rF)%r isinstancer list get_trialsrP study_namestater COMPLETErOr8rr system_attrsgetr allrrrvalue directionr MINIMIZErLrQminimum accumulatemaximumrr$_loggerwarningsumclearmax)r(r*r,r-studiestrialsr rrWtrial constraintsr'feasible_best_values best_valuesbest_label_namefeasible_trial_countinfeasible_trial_countall_trial_numbersr^eb_trial_numberseb_values_infoeb_best_values_inforIrr[r#_get_optimization_history_info_list/s                 rzObjective ValueF)r*r,r- 'go.Figure'cCs tt||||}t||S)a7Plot optimization history of all trials in a study. Args: study: A :class:`~optuna.study.Study` object whose trials are plotted for their target values. You can pass multiple studies if you want to compare those optimization histories. target: A function to specify the value to display. If it is :obj:`None` and ``study`` is being used for single-objective optimization, the objective values are plotted. .. note:: Specify this argument if ``study`` is being used for multi-objective optimization. target_name: Target's name to display on the axis label and the legend. error_bar: A flag to show the error bar. Returns: A :class:`plotly.graph_objects.Figure` object. )rcheckr_get_optimization_history_plot)r(r*r,r-r\rrrplot_optimization_historys rr\c Cstjdddid|id}g}|D]\}}}ddt||jD}|jdurQd}ddt||jD} g} | D] } | |j| q6g} |D] } | |j| qEntj|jvs]tj |jvrbt dd |jd d }|} |j} g} |tj | | |d |j d |dur|tj ||j|j dd|jdurt|jt|j} |tj || dtddddt|jt|j}|tj ||ddddd|tj || |d dddiddqtj||dS)NzOptimization History PlottitleTrial)rxaxisyaxiscSg|] \}}|tjkr|qSr)rr)r2rUr;rrrr4sz2_get_optimization_history_plot..cSrr)rr)r2numr;rrrr4s  z~Your study contains infeasible trials. In optimization history plot, error bars are calculated for only feasible trial values.dataT)typearrayvisiblemarkers)r3yerror_ymodenamelines)r3rrrg{Gz?)widthF)r3rrline showlegendnonetonextyzrgba(255,0,0,0.2))r3rrrfill fillcolorzInfeasible Trialcolorz#cccccc)r3rrrrmarkerr)rlayout)rLayoutrLr!rrOrrrrrnroScatterr rQrdictFigure)r\r,rtracesr%r&r'infeasible_trial_numbersrfeasible_trial_numbersfeasible_trial_valuesrinfeasible_trial_valuesupperlowerrrrrs          r) r(r)r*r+r,rr-r.r/r0) r(r)r*r+r,rr-r.r/r)r\r0r,rr/r)& __future__rcollections.abcrrenumrrMtypingrrnumpyrQoptuna.loggingroptuna.samplers._baser optuna.studyr optuna.study._study_directionr optuna.trialr r $optuna.visualization._plotly_importsroptuna.visualization._utilsr is_successfulrrrnrrr$rrrrrrrs:               "