o }oi`#@srddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddl m Z dd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZer|ddlmZddlmZddlmZddlmZe eZ GdddeZ!GdddeZ"GdddeZ#d=d'd(Z$d>d.d/Z% 0d?d0d1d2d@d4d5Z&dAd7d8Z'dBd;d<Z(d0S)C) annotations)Callable)Any)cast) NamedTuple)CategoricalChoiceType)CategoricalDistribution) get_logger)_CONSTRAINTS_KEY)Study) FrozenTrial) TrialState)_imports)_check_plot_args)_filter_nonfinite) _is_log_scale)go) make_subplots)Scatter) COLOR_SCALEc@sNeZdZUded<ded<ded<ded<d ed <d ed <d ed <ded<dS)_SliceSubplotInfostr param_name list[Any]x list[float]y list[int] trial_numbersboolis_log is_numericalz list[bool] constraints(tuple[CategoricalChoiceType, ...] | Nonex_labelsN__name__ __module__ __qualname____annotations__r*r*O/home/ubuntu/.local/lib/python3.10/site-packages/optuna/visualization/_slice.pyrs  rc@seZdZUded<ded<dS)_SlicePlotInfor target_namezlist[_SliceSubplotInfo]subplotsNr%r*r*r*r+r,)s  r,c@s&eZdZUded<ded<ded<dS) _PlotValuesrrrrrrNr%r*r*r*r+r/.s  r/trialslist[FrozenTrial]paramrtarget%Callable[[FrozenTrial], float] | None log_scaler numericalr$r#returnc Cs|dur d dd}|}t|ggg|||gd}|D]7}||jvr!q|j|j||j|||j|j|jt } |j | dupNt dd | Dq|S) Ntr r7floatcSs tt|jSN)rr9value)r8r*r*r+_target>s z(_get_slice_subplot_info.._target)rrrrr r!r$r"cSsg|]}|dkqS)gr*).0rr*r*r+ Usz+_get_slice_subplot_info..)r8r r7r9) rparamsrappendrrnumber system_attrsgetr r"all) r0r2r3r5r6r$r< plot_infor8r"r*r*r+_get_slice_subplot_info4s,   $rFstudyr r?list[str] | Noner-c st||t|jdtjfddtdkr#tdt|gSddD}iD]}|j D] \}}|vrA||<q5q.i D]\}}t |t rW|j |<qI|durat|}n|D]} | |vrqtd| d qctt|}t|fd d |Dd S) NF)deepcopystates)r3rz.Your study does not have any completed trials.cSs h|] }|jD]}|q qSr*)r?keys)r=r8p_namer*r*r+ js z'_get_slice_plot_info..z Parameter z does not exist in your study.c s8g|]}t|t|t|t |dqS))r0r2r3r5r6r$)rFr isinstancerrC)r=r2 distributionsr3r0r$r*r+r>s z(_get_slice_plot_info..)r-r.)rr get_trialsr COMPLETElen_loggerwarningr,rPitemsrNrchoicessorted ValueErrorset) rGr?r3r- all_paramstrialr distribution sorted_params input_p_namer*rOr+_get_slice_plot_infoZsB         r`NzObjective Value)r3r- 'go.Figure'cCsttt||||S)aNPlot the parameter relationship as slice plot in a study. Note that, if a parameter contains missing values, a trial with missing values is not plotted. Args: study: A :class:`~optuna.study.Study` object whose trials are plotted for their target values. params: Parameter list to visualize. The default is all parameters. 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. Returns: A :class:`plotly.graph_objects.Figure` object. )rcheck_get_slice_plotr`)rGr?r3r-r*r*r+ plot_slicesrdinfocCstjdd}t|jdkrtjg|dSt|jdkrZtjt|jd|d}|j|jdjd|j|j d|jdj sL|jdd|jdj d |S|jdj rX|jd d |St dt|jd d }||d }t|jddD]S\}}t|}|djd|id|rd}|D] }|j|d|dq|j|jd|d|dkr|j|j d|d|j s|jdd|j d|dqr|j r|jd d|dqrt|jdkr|jdt|jd|S)Nz Slice Plot)titler)datalayout) title_textcategoryarray)type categoryorder categoryarraylog)rmT)rowscols shared_yaxes)start showscale)markerF)rowcol)rjrwrx)rmrnrorwrx)rmrwrxi,)width)rLayoutrSr.Figure_generate_slice_subplot update_xaxesr update_yaxesr-r!r$r r update_layout enumerateupdate add_trace)rerhfigureru column_index subplot_infotracer8r*r*r+rcsT  !  rcr list[Scatter]cCsg}tggg}tggg}t|j|j|j|jD]7\}}}}|dus/|dks/|dus/|dkrP|rD|j||j||j|q|j||j|q|tj|j|jddddd|jt ddd d d d d t |jdkr|tj|j|jddddid d |S)NNonemarkerszFeasible Trialg?Grey)rzcolorTrialg?()rfrxpad)liner colorscalecolorbarF)rrmodenamerv showlegendrzInfeasible Trialrz#cccccc) r/ziprrrr"r@rrrrS)rrfeasible infeasiblerrnumcr*r*r+r}sX         r})r0r1r2rr3r4r5rr6rr$r#r7r) rGr r?rHr3r4r-rr7r,r:) rGr r?rHr3r4r-rr7ra)rer,r7ra)rrr7r)) __future__rcollections.abcrtypingrrroptuna.distributionsrroptuna.loggingr optuna.samplers._baser optuna.studyr optuna.trialr r $optuna.visualization._plotly_importsroptuna.visualization._utilsrrr is_successfulrrrrr&rTrr,r/rFr`rdrcr}r*r*r*r+sD                      &7 !.