L iiddlmZddlmZddlmZddlZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZej,rddlmZddlmZddlmZddlmZddlmZe d dddd ddZddZ ddZ ddZ d dZ y)!) annotations) defaultdict)CallableN)Any)experimental_func)Study) FrozenTrial)_get_slice_plot_info) _PlotValues)_SlicePlotInfo)_SliceSubplotInfo)_imports)Axes)Colormap) matplotlib)PathCollection)pltz2.2.0zObjective Value)target target_namecXtjtt||||S)aVPlot the parameter relationship as slice plot in a study with Matplotlib. .. seealso:: Please refer to :func:`optuna.visualization.plot_slice` for an example. 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:`matplotlib.axes.Axes` object. )rcheck_get_slice_plotr )studyparamsrrs l/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/optuna/visualization/matplotlib/_slice.py plot_slicers&@ NN /vv{S TTc nt|jdk(rtj\}}|Stjd}d}tjj dt|jdk(rOtj\}}|j dt|jd||||j}ntjdddz }tjdd} tjdt|jd |t|jz| f \}}|jdt|jD]#\} } || }t| ||||j}%|j| } | jd |S) NrBluesg?ggplotz Slice Plotzfigure.figsizeT)shareyfigsize)axTrial)lensubplotsrget_cmapstyleuse set_title_generate_slice_subplotrrrcParamssuptitle enumeratecolorbar set_label) info_r%cmap padding_ratiofigaxssc min_figwidthfighightisubplotaxcbs rrr>s} 4==Q 2  << DMIIMM( 4==Q<<>S l#%T]]1%5sD-QUQaQa b"**+; E2r2&HRR(J  (B(BLD\D\ 5 1c1 =AK1=AK !!!$ !!!$&&--c2 ##A& ##A&((//4 5 e  ZZ ZZ ++ !|| !|| -I,X`-a* J (D\S](^% lA !*|";]E RDKKQa! J jtPV WBJJ|\Y>PJQNN Irc|jJg}g}g}tt}t|j|j |j D]\}}}||j||f|jD]K} || D]A\}}|jt| |j||j|CM|||fSN) x_labelsrlistrKrLrMrNrPstr) rYvaluesvalue_xvalue_yvalue_c points_dictrLrMnumberx_labels rrTrTs  ,, ,GGGd#KFHHfhh0D0DE+ 1fAq&k*+((#$W- #IAv NN3w< ( NN1  NN6 " ## GW $$rct|}t|}|dk(rtj|tj|z |z}tjdtj||z tjdtj||zfS|dk(r$t t |dz }||z}| ||zfS||z |z}||z ||zfS)NrC rDr!)maxminmathlog10powr'rI)rir6rZ value_max value_minpaddingwidths rrUrUsF IF I ~::i(4::i+@@MQ HHRI.8 9 HHRI.8 9   - CK 1$-'x((y(M97"I$777rre) rrrzlist[str] | Nonerz%Callable[[FrozenTrial], float] | Nonerrhreturn'Axes')r3r r{r|) rYr r%r|r5z 'Colormap'r6floatrrhr{z'PathCollection')rYr rir r{z(tuple[list[Any], list[float], list[int]])riz list[Any]r6r}rZz str | Noner{ztuple[float, float])! __future__r collectionsrcollections.abcrrttypingroptuna._experimentalr optuna.studyr optuna.trialr optuna.visualization._slicer r r r 3optuna.visualization.matplotlib._matplotlib_importsr is_successfulrrrrrrrr-rTrUrrrs3"#$ 2$<369H8HLNRG7 $ U59( U  U  U 2 U  U  U UF&R+#++ + +  +  +\%#%-8%-%$8 8&+84>88r