bcg b d dl Z d dlZd dlZddlmZ ej
Zd Zd Z G d de Z y) N ) _catboostc ddl m} ddl m} |d t | t s| g} | D ]
} || y # t $ r0}d}t j
| t t
| d }~ww xY w)Nr )iplot)init_notebook_modeT) connected(To draw plots you should install plotly.) plotly.offliner r ImportErrorwarningswarnstr
isinstancelist)figsr r ewarn_msgfigs P/var/www/html/bid-api/venv/lib/python3.12/site-packages/catboost/plot_helpers.pytry_plot_offliner sk "(5T*
dD !v
c
"=
h #a&!!"s : A3+A..A3c 4 d} ddl m fd}t
| t
r't t | d 5 } || d d d y || y # t $ r.}t j | t t | d }~ww xY w# 1 sw Y y xY w)Nr r )plotc | j dj dddddj ddf D ]. } |d d
d
}| j d j | 0 | j d
y )N
zzzz@z{}zzdivF) output_type show_linkinclude_plotlyjsz
{}
z
)writejoinformat)plot_file_streamr graph_divr plot_name plotly_plots r write_plot_filez'save_plot_file..write_plot_file s tyy & N & &y 1 *
C# ! !& I
" "8??9#= > 12 w) r
r r r r
r r openfspath) plot_filer$ r r r r& r" r% s `` @r save_plot_filer, s 9H"6
3( )S!
&#S
)-= , - *
) "9 "
h #a&!!"2 *
)s" A B B )BB Bc * e Zd Zd Zd Zd Zd Zd Zy)OfflineMetricVisualizerc L t |t r|g}| j | y N)r r
_load_data)self
train_dirss r __init__z OfflineMetricVisualizer.__init__< s j# &$ J
#r' c g i d}t j j |d }t j j | r; t |d 5 }t j | }|d |d< |d |d< d d d |d r |d d d nd|d r |d d |d
S d|d
S # 1 sw Y /xY w# t $ r Y =w xY w) N)
iterationsmetazcatboost_training.jsonrr7 r6 iterationr iteration_count)passed_iterationstotal_iterationsdata)ospathr isfiler) jsonload
ValueError)r2 r@ r>
training_json json_data
training_datas r _update_data_from_dirz-OfflineMetricVisualizer._update_data_from_dirA s
T+CD
77>>- (
--$(IIi$8M#0#8DL)6|)DD & . IM\HZl!3B!7 !D`aCG<V -> ?
UV
.-
s* B2 &B&8B2 &B/+B2 2 B>=B>c g }t j j t j j }|D ]V }t j j | }|j ||k7 r t j j | nd X i }t
|| D cg c]
\ }}||d } }}| D ]? }
|
j d }| j | } | s(||
j d | d||< A || _ y c c}}w )Ncurrent)namer@ r@ rK )r@ rK content)
r? r@ abspathcurdirappendbasenamezipgetrH r> ) r2 r3 namesrN train_dirrM r> rK r@ dirsdir_inforL s r r1 z"OfflineMetricVisualizer._load_dataX s 0#Iggooi0G LLg6G))'2Y W $ ?B5*?UV?Utt,?UV H<<'D006G V,"DJ Ws &Dc dd l m} g }| j j D ]k \ }}|d d d }|d dk( }i } t |d D ]7 \ }
} | d } | j | i |d D
cg c] }
|
|
f c}
| | d < 9 t |d
D ]7 \ }
}|d } | j | i |d D
cg c] }
|
|
f c}
| | d < 9 |d d d }| j D ] \ } }|j }|d d
k( r|n|d dz | z }|j j | |d d< d}|j d g D ]i \ }}|j |j |D cg c] }|d c}|D cg c]
}|| | c}|j j# | d| k d}|j d g D ] \ }}d }|j |j |D cg c] }|d c}|D cg c]
}|| | c}|j j# | d||r|D cg c]
} || c}nd |j% t' d t' | |j) | n |S # t $ r.}t j d t t | d }~ww xY wc c}
w c c}
w c c}w c c}w c c}w c c}w c c}w ) Nr z1To save plots to files you should install plotly.rL r> r7
parameters
learn_metricsrK
learn_sets test_metrics test_setsr6
catboost_info : )textlayouttitlez rgb(160,0,0)r: )colorlines)xylinemoderK z rgb(0,160,0)c g }| d d j D ]% \ }}|j |dz t | z ' dj | S )NrX r r^ z
)itemsrO r r )r result
param_name param_values r generate_params_hoverzGOfflineMetricVisualizer._get_plotly_figs..generate_params_hover sS !#78 q7I7O7O7Q 3J "MM*u* ri enumerate
setdefaultFigurer` TitlerR add_traceScatterscatterLine
update_layoutdictrO )r2 ra goerrr r@ dir_datar7 has_parametersmetricsi learn_metric metric_nameset_name test_metricr6 subsetsr figure_titlelearn_graph_colorlearn_set_name
metric_idxr test_graph_color
test_set_namerm s r _get_plotly_figsz(OfflineMetricVisualizer._get_plotly_figsp sK ( *
"iioo /ND(I&v.v6D !, /< ?N G#,T/-B#C< *6 2 "";3TXYeTf5gTfxmTf5g $\2 $D #,D ,@"A; )& 1 "";3SWXcSd4eSdxh]Sd4e $[1 #B
"),V4\BJ(/
$ Wiik*26*:o*MT\]cTdhm mp{ { )+ l )KH
g&$2!29++lB2O.NJMM"**3=>:a1[>:>BLM*Q1^,Z8*M ZZ__3D_ E $ + #- # 3P $2 18["1M-M:3 MM"**3=>:a1[>:>AKLA1]+J7L ZZ__3C_ D $ *TbZ"PZ#8#;Z"Phl
#- # 2N" !! \2 [1 "
C S )8% 0z E ( MMM N c#h' ' ( 6h 5f ?M" ?L #QsA J 2 K: K- K?K 0 K#K(6K- K )KK c < t ||| j | y r0 )r, r )r2 ra file_outputs r save_to_filez$OfflineMetricVisualizer.save_to_file s {E4+@+@+GHr' N)__name__
__module__ __qualname__r4 rH r1 r r r' r r. r. ; s $
.0ENIr' r. )
rB r? r r r* r r, objectr. r r' r r s7 #F}If }Ir'