2VhE(dZddlZddlZddlmZddlmZddlmZ ddl Z dZ dZdZd Zd Zd Zed  dd Zed ddZy#e $r' ddl Z n#e $r ddl Z n #e $rdZ YnwxYwYnwxYwYnwxYw)z)Utilities related to model visualization.N)tree) keras_export)io_utilsctduS)z#Returns True if PyDot is available.N)pydotS/home/dcms/DCMS/lib/python3.12/site-packages/keras/src/utils/model_visualization.py check_pydotr s  r ctsy tjjtjy#ttj f$rYywxYw)z6Returns True if both PyDot and Graphviz are available.FT)r rDotcreateOSErrorPydotExceptionrr r check_graphvizrsI =  % U)) *s1?AActt|}tt|}|j||s:tj||}|j dd|j |yy)Npenwidth2)stridget_edgerEdgesetadd_edge)dotsrcdstsrc_iddst_idedges r rr*sY C\F C\F << 'zz&&) S! T (r ct|jdr|jj}|St|jdr|jj}|St |j}|S)Nname__name__)hasattr activationr"r#r)layeractivation_names r get_layer_activation_namer(3snu(**//  !!: .**33 e../ r c |jj}|jd}|jd}|jd}|jd}|jd}|rtd|d}t d||fD} |r| d z } t d | } |r|d | d |j d |dz }n |d | d |dz }|r-t|dr!|j|d| dt|dz }g} |rd} d} tjd|j} tjd|j} d}|dk7r| jd|| d| jd|| d|r=d} tjd|j}| jd|xsdd|rGt|dr;|j r/|j"r| jdn| jd| r t%| } nd } | r|d d!j'| zd"zz }|d#z }|S#ttf$rY wxYw#ttf$rYwxYw)$Nshow_layer_namesshow_layer_activations show_dtype show_shapesshow_trainablezInvalid kwargs: zC<c32K|]}t|ywN)int).0xs r z#make_layer_label..LsCc!fCszzr%z!z;z\zr?zz
z (z)
Activation: c|jSr0shaper3s r z"make_layer_label..ps qwwr c|jSr0r8r:s r r;z"make_layer_label..qs r c |ot|tr8dj|jDcgc] \}}|d|c}}}n|}|j ddj dd}|Sd}|Scc}}w)Nz, z: }{?) isinstancedictjoinitemsreplace)r9kv shape_strs r format_shapez&make_layer_label..format_shapeus eT* $ 16?AA3b?!I$)'I%--c26>>sBG    @sA6 InputLayerz:Input shape: zOutput shape: c|jSr0)dtyper:s r r;z"make_layer_label..s r z;Output dtype: rA trainablezZTrainableNon-trainable
>) __class__r#pop ValueErrorsummaxr"r$r%r(r map_structureinputoutputAttributeErrorappendweightsrNlenrD)r&kwargs class_namer*r+r,r-r.table colspan_maxcolspancols input_shape output_shaperJrMs r make_layer_labelrc=s))Jzz"45#ZZ(@AL)J**]+KZZ 01N +F8455 N Cz>&BCCKq ![!G y)**VJ<0    y)     E< (    ( /y97>?@    D   ,,-> LK--.?NL   % KK''3K'@&AB##  $$0$>#?@  &&'8%,,GE $$)LS>2 '%5%-- ?? KK3  KK7  d)  "''$-''11 [E LKN+   FN+   s%AI !IIII('I(c tjtt|t |fi|}|j dd|j dd|j dd|S)N)labelfontname Helveticaborder0margin)rNoderrrcr)r&r[nodes r make_nodermsT ::c"U)n,D]i\}}|j@}|jB |jBj:|jD}|jF}|jB}|}|st9|||mt!|| j&| j"fr@|jH|j@\}}}t!|| j&| j"fr@t!|| j&| j"frlt!|| j&r$|jJ|j@jB}n|j$d}t!|| j&| j"frlt9|||l|S)aConvert a Keras model to dot format. Args: model: A Keras model instance. show_shapes: whether to display shape information. show_dtype: whether to display layer dtypes. show_layer_names: whether to display layer names. rankdir: `rankdir` argument passed to PyDot, a string specifying the format of the plot: `"TB"` creates a vertical plot; `"LR"` creates a horizontal plot. expand_nested: whether to expand nested Functional models into clusters. dpi: Image resolution in dots per inch. subgraph: whether to return a `pydot.Cluster` instance. show_layer_activations: Display layer activations (only for layers that have an `activation` property). show_trainable: whether to display if a layer is trainable. Returns: A `pydot.Dot` instance representing the Keras model or a `pydot.Cluster` instance representing nested model if `subgraph=True`. r) make_node_keyyThis model has not yet been built. Build the model first by calling `build()` or by calling the model on a batch of data.) functional) sequentialzFYou must install pydot (`pip install pydot`) for model_to_dot to work.dashed)style graph_namere labeljustlrankdir concentrateTdpisplinesorthorecordr8 layer_rangeN.Argument `layer_range` is no longer supported. Unrecognized keyword arguments: )r*r+r,r-r.)subgraphr+r.r6)&keras.src.ops.functionrobuiltrQkeras.src.modelsrqrrr ImportErrorrClusterr"rr set_node_defaultsrPrB Sequentiallayers Functionalrmadd_node _operations enumerate model_to_dot add_subgraphrangerZr_inbound_nodes_nodes input_tensors_keras_history operation node_index tensor_indexoutputsinputs)modelr-r,r*rx expand_nestedrzrr+r.r[rorqrrrrrlir&submodel inbound_index inbound_node input_index input_tensor input_history input_node output_indexsource destination_s r rrsOJ5 ;; ,  ,+ = $  mm(uzzB $  S!iik  7#  t$ s  7# H- zz-&2IJJ ;F8DEE-"8 "( F%../ z44 5)&) T ""f%5 Z J)):+@+@A $ '=- H   X &U-f-D LL ).%../3v;?+ 8fQiA7 8J \\!"%FM3+4U5I5I+JK 3 'M<5*"7"78!%7u||K.7**.B 3) \ !- ; ;  **2+44CC!,,  -99 $--# $S&%0 !**J,A,AB/5nn$/$n,FA|!**J,A,AB$!**J,A,AB"+z/D/DE'2&8&8''(.$'2&8&8&; #!**J,A,AB&fk2EB 3K 3M3\ Jr zkeras.utils.plot_modelc |js tdts5d} dtjvrt j | yt| ts5d} dtjvrt j | yt| | jdd td| rtd| t||||||||| } t|}| ytjj|\} }|sd }n|d d}| j|| |d k7r ddlm}|j%|Sy#t$rYywxYw)aConverts a Keras model to dot format and save to a file. Example: ```python inputs = ... outputs = ... model = keras.Model(inputs=inputs, outputs=outputs) dot_img_file = '/tmp/model_1.png' keras.utils.plot_model(model, to_file=dot_img_file, show_shapes=True) ``` Args: model: A Keras model instance to_file: File name of the plot image. show_shapes: whether to display shape information. show_dtype: whether to display layer dtypes. show_layer_names: whether to display layer names. rankdir: `rankdir` argument passed to PyDot, a string specifying the format of the plot: `"TB"` creates a vertical plot; `"LR"` creates a horizontal plot. expand_nested: whether to expand nested Functional models into clusters. dpi: Image resolution in dots per inch. show_layer_activations: Display layer activations (only for layers that have an `activation` property). show_trainable: whether to display if a layer is trainable. Returns: A Jupyter notebook Image object if Jupyter is installed. This enables in-line display of the model plots in notebooks. rpzFYou must install pydot (`pip install pydot`) for `plot_model` to work.zIPython.core.magics.namespaceNznYou must install graphviz (see instructions at https://graphviz.gitlab.io/download/) for `plot_model` to work.r~rr)r-r,r*rxrrzr+r.pngr6)formatpdfr)display)filename)rrQr sysmodulesr print_msgrrrPrrospathsplitextwriteIPythonrImage)rto_filer-r,r*rxrrzr+r.r[messagerr extensionrs r plot_modelrsy` ;; ,  = (  +ckk 9   w ' g& &   (  +ckk 9   w ' g& & zz-&2IJJ ;F8DEE  )# 5% C'lG {77##G,LAy  abM IIgiI(E  '=='=2 2     s*E EE) FFTTBFFFF) z model.pngFFFrFrFF)__doc__rr keras.srcrkeras.src.api_exportrkeras.src.utilsrrrpydot_ng pydotplusr rrr(rcrmrrrr r rs/ -$  zz()   O*Od&'    r(rU     % E  sYA%%B+A0/B0B 6A;:B ;BB BB B B  BB