L i> d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlZd dl Z d dl m
Z
d dl m Z d d l
mZ d d
lmZ d d lmZ d d lmZ d d
lmZ d dlmZ ej0 rd dlmZ e j4 j7 e Z G d de Z dddddd ddZ d dZ d d dZ!d dZ" d dZ# d! d"dZ$ d! d#dZ%y)$ ) annotations)Callable)Sequence)Any)
NamedTupleN) _deprecated)_CONSTRAINTS_KEY)Study)"_get_pareto_front_trials_by_trials) FrozenTrial)
TrialState)_imports)_make_hovertext)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__ h/mnt/ssd/data/python-lab/Trading/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)r/
infeasiblez$%{text}Feasible Trial zPareto-front Plotr )title xaxis_title yaxis_title)r5 r6 zaxis_title)r4 scene)datalayout) r r _make_scatter_objectr r r r r r Layoutr Figure)r, r r r9 r: s r$ r* r* m s # < < **O (00 ;!%
!(,, @!&
(
!(22 F
!(00 D!%
!(,, @!&
#
6 ~~% ))$//!*<= ))$//!*<=
% # 0 0 1C D # 0 0 1C D # 0 0 1C D
99$v ..r# c
|z(_get_pareto_front_info.. s
c r# c 3 & K | ] }|d k yw)rE Nr" ).0rF s r$ z)_get_pareto_front_info.. s %D1a3h%Ds r completedzcompleted and feasiblez Your study does not have any z trials. )r2 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!.r3 zGiven `axis_order` z contains invalid index z
higher than z lower than 0.)r r r r r r r r )rS list[FrozenTrial]r& z(Callable[[FrozenTrial], Sequence[float]]returnr )rZ -Sequence[tuple[FrozenTrial, Sequence[float]]]r^ z
int | None) r _DEPRECATION_WARNING_TEMPLATEformatwarningswarn
FutureWarningrO
get_trialsr
COMPLETEallmapappend system_attrsgetr r
directions _get_non_pareto_front_trialsrY _loggerwarning_targets_default metric_namesrangerR setmaxminr )r+ r r r r% r& msgfeasible_trialsinfeasible_trialsr rT constraints best_trialsnon_best_trials
what_trial_targetsrW r r r r[ r rq is r$ r) r) sW 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 $3[9Q
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 = Nc | j S N)values)rT s r$ rp rp K s <<r# c D g }| D ] }||vs|j | |S r )ri )rS
pareto_trialsnon_pareto_trialsrT s r$ rm rm O 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)r0 r1 r2 r r3 markersF)rF ytextmoder/ marker
showlegendrL )rF r zr r r/ r r zMust not reach here) _make_markerr Scatterr Scatter3d) r r r rZ r/ r1 r0 rT _r r s r$ r; r; Y 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 Bluesr4 Trial)liner
colorscalecolorbarRedsz
Best Trialg?r3 ( )r4 rF xpad)rY number)rS r r0 r1 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 )rT r r^ zSequence[float])rS r] r r] r^ r] )FF)r r r z
Sequence[int]r r rZ r_ r/ strr1 r r0 r r^ z 'go.Scatter' | 'go.Scatter3d')
rS zSequence[FrozenTrial]r r r0 r r1 r r^ zdict[str, Any])&
__future__r collections.abcr r typingr r rb optunar optuna.samplers._baser optuna.studyr
optuna.study._multi_objectiver optuna.trialr r
$optuna.visualization._plotly_importsr optuna.visualization._utilsr
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#