:j;? d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl Z d dl m
Z
d d l m Z d d
l
mZ d d lmZ d d lmZ d d
lmZ d dlmZ d dlmZ erd dlmZ ej4 rd dlmZ e j8 j; e Z G d de Z dddddd d dZ!d dZ" d d dZ#d!dZ$ d"dZ% d# d$dZ& d# d%d Z'y)& ) annotations)Callable)Sequence)Any)
NamedTuple)
TYPE_CHECKINGN) _deprecated) optuna_warn)_CONSTRAINTS_KEY)"_get_pareto_front_trials_by_trials) FrozenTrial)
TrialState)_imports)_make_hovertext)Study)goc ^ e Zd ZU ded< ded< ded< ded< ded< d ed
< d ed < d ed
< y)_ParetoFrontInfoint n_targetsz list[str] target_names%list[tuple[FrozenTrial, list[float]]]best_trials_with_valuesnon_best_trials_with_values infeasible_trials_with_valuesz list[int]
axis_orderboolinclude_dominated_trialshas_constraintsN)__name__
__module__ __qualname____annotations__ w/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/optuna/visualization/_pareto_front.pyr r s1 N BB!FF#HH ""r% r T)r r r constraints_functargetsc ` t j t | ||||| }t | S )a Plot the Pareto front of a study.
.. seealso::
Please refer to :ref:`multi_objective` for the tutorial of the Pareto front visualization.
Args:
study:
A :class:`~optuna.study.Study` object whose trials are plotted for their objective
values. The number of objectives must be either 2 or 3 when ``targets`` is :obj:`None`.
target_names:
Objective name list used as the axis titles. If :obj:`None` is specified,
"Objective {objective_index}" is used instead. If ``targets`` is specified
for a study that does not contain any completed trial,
``target_name`` must be specified.
include_dominated_trials:
A flag to include all dominated trial's objective values.
axis_order:
A list of indices indicating the axis order. If :obj:`None` is specified,
default order is used. ``axis_order`` and ``targets`` cannot be used at the same time.
.. warning::
Deprecated in v3.0.0. This feature will be removed in the future. The removal of
this feature is currently scheduled for v5.0.0, but this schedule is subject to
change. See https://github.com/optuna/optuna/releases/tag/v3.0.0.
constraints_func:
An optional function that computes the objective constraints. It must take a
:class:`~optuna.trial.FrozenTrial` and return the constraints. The return value must
be a sequence of :obj:`float` s. A value strictly larger than 0 means that a
constraint is violated. A value equal to or smaller than 0 is considered feasible.
This specification is the same as in, for example,
:class:`~optuna.samplers.NSGAIISampler`.
If given, trials are classified into three categories: feasible and best, feasible but
non-best, and infeasible. Categories are shown in different colors. Here, whether a
trial is best (on Pareto front) or not is determined ignoring all infeasible trials.
.. warning::
Deprecated in v4.0.0. This feature will be removed in the future. The removal of
this feature is currently scheduled for v6.0.0, but this schedule is subject to
change. See https://github.com/optuna/optuna/releases/tag/v4.0.0.
targets:
A function that returns targets values to display.
The argument to this function is :class:`~optuna.trial.FrozenTrial`.
``axis_order`` and ``targets`` cannot be used at the same time.
If ``study.n_objectives`` is neither 2 nor 3, ``targets`` must be specified.
.. note::
Added in v3.0.0 as an experimental feature. The interface may change in newer
versions without prior notice.
See https://github.com/optuna/optuna/releases/tag/v3.0.0.
Returns:
A :class:`plotly.graph_objects.Figure` object.
)r check_get_pareto_front_info_get_pareto_front_plot)studyr r r r' r( infos r& plot_pareto_frontr/ ) s6 @
NN !
| 5zCSU\ D "$ ''r% c
| j }| j }|s_t | j | j || j
dd t | j | j || j dd g}nt | j | j || j dd t | j | j || j
dd t | j | j || j dd g}| j d k( rMt j d
| j | j d | j | j d
}nit j d
| j | j d | j | j d | j | j d d }t j || S )Nz%{text}TrialT)
hovertemplatedominated_trialsz %{text}Best TrialFz&%{text}Infeasible Trial)r1
infeasiblez$%{text}Feasible Trial zPareto-front Plotr )title xaxis_title yaxis_title)r7 r8 zaxis_title)r6 scene)datalayout) r r _make_scatter_objectr r r r r r Layoutr Figure)r. r r r; r< s r& r, r, q s # < < **O (00 ;!%
!(,, @!&
(
!(22 F
!(00 D!%
!(,, @!&
#
6 ~~% ))$//!*<= ))$//!*<=
% # 0 0 1C D # 0 0 1C D # 0 0 1C D
99$v ..r% c
|2t j j ddd }t |t |2t j j ddd }t |t |
| t d g }g }d } | j
d t j f
D ] }
|Bd } t t d ||
r|j |
n|j |
G|
j j t } | | d uz } | t d
| D r|j |
|j |
t || j } |r
t# || }
ng }
t% | dk( r | rdnd}t& j) d| d |}|)t% | j dv rt* }n t d d(d} || | } ||
| } ||| } d)d} || xs || }|1| t% | }n#|t% | j }n t d |dvrt d| d |-| j, }| t/ | D cg c] }d| }}n |}n t% | |k7 rt d| d |t1 t/ | }nt% | |k7 r t d | d | d t% | d t% t3 | |k7 rt d | d! t5 | |d"z
kD r!t d#| d$t5 | d%|d"z
d t7 | dk rt d#| d$t7 | d& t9 |||||||| ' S c c}w )*Nz `axis_order`z3.0.0z5.0.0)named_verr_verz`constraints_func`z4.0.0z6.0.0z]Using both `targets` and `axis_order` is not supported. Use either `targets` or `axis_order`.F)deepcopystatesTc | dk S )N r$ )xs r& z(_get_pareto_front_info.. s
c r% c 3 & K | ] }|d k yw)rG Nr$ ).0rH s r& z)_get_pareto_front_info.. s %D1a3h%Ds r zcompleted and feasible completedz Your study does not have any z trials. )r4 z`plot_pareto_front` function only supports 2 or 3 objective studies when using `targets` is `None`. Please use `targets` if your objective studies have more than 3 objectives.c | D cg c]
} || }}|D ]) }t |t rt dt | t | | D cg c] \ }}|t | f c}}S c c}w c c}}w )NzL`targets` should return a sequence of target values. your `targets` returns )
isinstancer
ValueErrortypeziplist)trialsr( trial
target_valuesvs r& _make_trials_with_valuesz8_get_pareto_front_info.._make_trials_with_values s 6<._infer_n_targets s+ ! "Q &)!,Q/0 0r% zNIf `targets` is specified for empty studies, `target_names` must be specified.zD`plot_pareto_front` function only supports 2 or 3 targets. you used z
targets now.z
Objective z/The length of `target_names` is supposed to be .zSize of `axis_order` z
. Expect: z
, Actual: z Elements of given `axis_order` z are not unique!.r5 zGiven `axis_order` z contains invalid index z
higher than z lower than 0.)r r r r r r r r )rU list[FrozenTrial]r( z(Callable[[FrozenTrial], Sequence[float]]returnr )r\ -Sequence[tuple[FrozenTrial, Sequence[float]]]r` z
int | None) r _DEPRECATION_WARNING_TEMPLATEformatr
FutureWarningrQ
get_trialsr COMPLETEallmapappend system_attrsgetr r
directions _get_non_pareto_front_trialsr[ _loggerwarning_targets_default metric_namesrangerT setmaxminr )r- r r r r' r( msgfeasible_trialsinfeasible_trialsr rV constraints best_trialsnon_best_trials
what_trial_targetsrY r r r r] r rq is r& r+ r+ sO 77>> wg ?
C '#77>>%WG ?
C 'z 5
4
OO!!5*:M:M9O!P , ' "O3)+;E+BCD &&u-!((/ ((,,-=> ;d22 #%D %D"D " "5 ) $ $U + ," 5_eFVFVWK 6 T
;1 1@-k
7
|9MNH u F *'H J
M! M9 M
/ M 7{H M":?H"U $<=NPX$Y! I !!89 =M%>I #L)I
_E , ,-I `
" =
2
)) 6;I6FGj ,GLG'L \ i 'J9+UVWXX% *+
z?i ' '
|:i[ I z?+1.
s: 9 , >zlJ[ \] ] z?Y] * %j\1I#j/IZ [ (1}oQ0
z?Q %j\1I#j/IZ [
! 7$?&C !9' 7 Hs ) M/c | j S N)values)rV s r& rp rp O s <<r% c D g }| D ] }||vs|j | |S r )ri )rU
pareto_trialsnon_pareto_trialsrV s r& rm rm S s7 ,
% $ $U +, r% c |xs g }t |D cg c] \ }}| c}}||| } | dk( rjt j |D
cg c]
\ }}
|
|d c}
}|D
cg c]
\ }}
|
|d c}
}|D cg c] \ }}t | c}}d|| d S | dk( rt j |D
cg c]
\ }}
|
|d c}
}|D
cg c]
\ }}
|
|d c}
}|D
cg c]
\ }}
|
|d c}
}|D cg c] \ }}t | c}}d|| d S J d
c c}}w c c}
}w c c}
}w c c}}w c c}
}w c c}
}w c c}
}w c c}}w ) N)r2 r3 r4 r r5 markersF)rH ytextmoder1 marker
showlegendrN )rH r zr r r1 r r zMust not reach here) _make_markerr Scatterr Scatter3d) r r r r\ r1 r3 r2 rV _r r s r& r= r= ] sb ,1r
1285!2 ) F A~zz6HIFvjm$I6HIFvjm$I9KLXUA/%(L'
a ||6HIFvjm$I6HIFvjm$I6HIFvjm$I9KLXUA/%(L'
,++u7 3 JIL JIILs/ D(
D.
D4 6D:1E
E %E ?Ec |r|st | dk( sJ |rddiS |r$ddd| D cg c] }|j c}ddd id
S ddd| D cg c] }|j c}d d |rd
ndddd
S c c}w c c}w )Nr colorz#ccccccg ?Grey)widthr Bluesr6 Trial)liner
colorscalecolorbarRedsz
Best Trialg?r5 ( )r6 rH xpad)r[ number)rU r r2 r3 ts r& r r s 86{a Y
!F3(./1ahh/!
"F3(./1ahh/ % 4S!
0 0s A+ A0)r- r r list[str] | Noner r r list[int] | Noner' /Callable[[FrozenTrial], Sequence[float]] | Noner( r r` 'go.Figure')r. r r` r )NTNNN)r- r r r r r r r r' r r( r r` r )rV r
r` zSequence[float])rU r_ r r_ r` r_ )FF)r r r z
Sequence[int]r r r\ ra r1 strr3 r r2 r r` z 'go.Scatter' | 'go.Scatter3d')
rU zSequence[FrozenTrial]r r r2 r r3 r r` zdict[str, Any])(
__future__r collections.abcr r typingr r r optunar optuna._warningsr
optuna.samplers._baser optuna.study._multi_objectiver optuna.trialr
r $optuna.visualization._plotly_importsr optuna.visualization._utilsr optuna.studyr
is_successfulr logging
get_loggerr rn r r/ r, r+ rp rm r= r r$ r% r& r s " $ $
( 2 L $ # 9 7 " 87
..
#
#H
-z &*%)#'HL?CE( E( #E( # E(
! E( F
E( =E( E(PA/L &*%)#'HL?C
W W"W #W ! W
F W =
W Wt .? "',', ', #', F ',
',
', ', #',Z #
!
"
r%