o
biE @ s d Z ddlZddlZddlmZ ddlmZ ddlmZ zddl Z W n+ e
yM zddl Z W n e
yJ zddl Z W n e
yG dZ Y nw Y nw Y nw dd Z
dd Zd
d Zd d
Zdd Zdd Zed d ddZed d dd ZdS ) z)Utilities related to model visualization. N)tree) keras_export)io_utilsc C s t duS )z#Returns True if PyDot is available.N)pydot r r W/home/ubuntu/.local/lib/python3.10/site-packages/keras/src/utils/model_visualization.py check_pydot s r c C s< t sdS z tjt W dS ttjfy Y dS w )z6Returns True if both PyDot and Graphviz are available.FT)r r DotcreateOSErrorPydotExceptionr r r r check_graphviz s r
c C sN t t|}t t|}| ||s%t||}|dd | | d S d S )Npenwidth2)stridget_edger Edgesetadd_edge)dotsrcdstsrc_iddst_idedger r r r * s r c C s> t | jdr | jj}|S t | jdr| jj}|S t| j}|S )Nname__name__)hasattr
activationr r r )layeractivation_namer r r get_layer_activation_name3 s
r" c K s* | j j}|d}|d}|d}|d}|d}|r&td| d}tdd ||fD } |r9| d
7 } td | }
|rP|d |
d
| j d| d7 }n |d |
d
| d7 }|rtt| drt| jd urt|d|
dt | d7 }g } |rd } d }
zt
dd | j } t
dd | j
}
W n ttfy Y nw dd }|dkr| d|| d | d ||
d |rd }z t
d d | j
}W n ttfy Y nw | d |pd d |rt| d r| jr| jr| d! n| d" | rt| }
nd }
| r|d#d$| d%7 }|d&7 }|S )'Nshow_layer_namesshow_layer_activations
show_dtype show_shapesshow_trainablezInvalid kwargs: zC<
c s s | ]}t |V qd S N)int).0xr r r L s z#make_layer_label.. z
z (z)
zr z!
Activation: c S | j S r( shaper+ r r r p z"make_layer_label..c S r/ r( r0 r2 r r r r3 q r4 c S sR | d ur%t | trddd | D }n| }|dddd}|S d}|S )Nz, c S s g | ] \}}| d | qS )z: r )r* kvr r r
y s z:make_layer_label..format_shape..} {?)
isinstancedictjoinitemsreplace)r1 shape_strr r r format_shapeu s
z&make_layer_label..format_shape
InputLayerz:
Input shape: z
z;
Output shape: c S r/ r( )dtyper2 r r r r3 r4 z;
Output dtype: r; trainablezZ
Trainable
z\
Non-trainable
z
r9 z
z
>) __class__r pop
ValueErrorsummaxr r r r" r
map_structureinputoutputAttributeErrorappendweightsrE lenr> )r kwargs
class_namer# r$ r% r& r' table colspan_maxcolspancols input_shape output_shaperB rD r r r make_layer_label= s
rZ c K sJ t jtt| t| fi |d}|dd |dd |dd |S )N)labelfontname Helveticaborder0margin)r Noder r rZ r )r rR noder r r make_node s
" rc zkeras.utils.model_to_dotFTTB c
K s$ ddl m} | js
tdddlm} ddlm}
t s td|r8t j
d| j d}| d | j | d
d n"t
}| d | | d
d | d| | dd |jdd |
dddurftd|
rotd|
||||| d}
t| |
jr| j}nt| | jst| fi |
}|| |S | j}t|D ]/\}}|rt|| j|
jfrt||||||d|| d }|| qt|fi |
}|| qt| |
jr|stt|d D ]}t ||| ||d q|S | jdd }|D ]}t|j D ]\}}t| | jr| ||| j vrqt|j D ]u\}}|j }|j!du r'q|j!j |j" }|j#}|j!}|} |sAt ||| qt|| j|
jfr^|j$| j \}} }t|| j|
jfsKt| | j|
jfrt| | jrw| j%| j j!} n| jd } t| | j|
jfsht ||| qqq|S )a Convert 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_namer[ labeljustlrankdir concentrateTdpisplinesorthorecordr0 layer_rangeN.Argument `layer_range` is no longer supported. Unrecognized keyword arguments: )r# r$ r% r& r' )subgraphr$ r' r. )&keras.src.ops.functionrf builtrH keras.src.modelsrh ri r ImportErrorr Clusterr r r set_node_defaultsrG r<
Sequentiallayers
Functionalrc add_node _operations enumerate model_to_dot add_subgraphrangerQ r _inbound_nodes_nodes
input_tensors_keras_history operation
node_index tensor_indexoutputsinputs) modelr& r% r# ro
expand_nestedrq rx r$ r' rR rf rh ri r r rb ir submodel
inbound_index inbound_node input_index input_tensor
input_history
input_node output_indexsource destination_r r r r s %
Lr zkeras.utils.plot_model model.pngc
K s* | j stdt s d} dtjv rt| dS t| t s1d} dtjv r-t| dS t| |
dddur=td|
rFtd|
t
| |||||||| d } t |}| du r]dS t j
|\}
}|sjd
}n|d d }| j||d |d
krz
ddlm} |j|dW S ty Y dS w dS )a Converts 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.
rg zFYou must install pydot (`pip install pydot`) for `plot_model` to work.z IPython.core.magics.namespaceNznYou must install graphviz (see instructions at https://graphviz.gitlab.io/download/) for `plot_model` to work.ru rv rw )r& r% r# ro r rq r$ r' pngr. )formatpdfr )display)filename)rz rH r sysmodulesr print_msgr| r
rG r r ospathsplitextwriteIPythonr Image)r to_filer& r% r# ro r rq r$ r' rR messager r extensionr r r r
plot_model sb 0
r ) FFTrd Fre FFF) r FFFrd Fre FF)__doc__r r keras.srcr keras.src.api_exportr keras.src.utilsr r r| pydot_ng pydotplusr r
r r" rZ rc r r r r r r sd
} S