B ӻd/@sdZddlZddlZddlmZddlmZddlmZy ddl Z Wn\e k ry ddl Z Wn:e k ry ddl Z Wne k rdZ YnXYnXYnXddZ dd Zd d Zed dddZeddddZdS)z)Utilities related to model visualization.N)path_to_string)nest) keras_exportc Cs@tdkr dSytjtdSttjfk r:dSXdS)z1Returns True if PyDot and Graphviz are available.NFT)pydotDotcreateOSErrorZInvocationExceptionr r Y/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/keras/utils/vis_utils.py check_pydot)sr cCs2ddlm}ddlm}t||jo0t|j|jS)Nr) functional)wrappers)tensorflow.python.keras.enginer tensorflow.python.keras.layersr isinstanceWrapperlayer Functional)rr r r r r is_wrapped_model6s   rcCs"|||s|t||dS)N)Zget_edgeadd_edgerEdge)dotsrcdstr r r r=s rzkeras.utils.model_to_dotFTTB`c' s$ddlm}ddlm} ddlm} tsLd} dtjkrDt| dSt | |r|t j d|j d } | d |j | d d n8t } | d || dd| d|| jddi} i}i}i}|j}|jst jtt||j d}| || St|| jr"|js|t| j|j}xt|D]\}}tt|}|j }|jj}t||jr|rt|j| j rt!|j|||||dd}|"}|d||jj <|d||jj <| #|n&d$||jj }|jjj}d$||}|r@t|| j r@t!||||||dd}|"}|d| |j <|d||j <| #||rTd$||}n|}|rxdd}d|||j%f}|rddy|j&}Wnt'k rd}YnXt(|drȈ|j)} n,t(|drd *fd!d"|j+D} nd} d#|| |f}|rt|| j s.t j||d}| |q.Wx|D]}tt|}xt|j,D]\}}|j d$t|}!|!|j-kr\xt./|j0D]z}"tt|"}#|s| 1|#st2| 1|st2t3| |#|qt|"| j st4|"st|| j s:t4|s:| 1|#st2| 1|s,t2t3| |#|nXt|| j r`t3| |#| |j 5n2t4|rt3| |#|||jj 5}$t3| ||$nzt|"| j r||"j 5}$t|| j r| |j 5}%t3| |$|%n t3| |$|n&t4|"r|"jj }&t3| ||&5|qWq\Wq.format_dtypez%s|%scSst|tddS)NNone)r'replace)r#r r r format_shapesz"model_to_dot..format_shaper& input_shape input_shapesz, csg|] }|qSr r ).0Zishape)r,r r sz model_to_dot..z %s |{input:|output:}|{{%s}|{%s}}z_ib-)6rr rrr r sysmodulesprint ImportErrorrZClusternamesetrZset_node_defaultslayersZ_is_graph_networkNoder'idadd_noderZ SequentialZbuiltbuildsuper enumerate __class____name__rrr model_to_dotZ get_nodesZ add_subgraphformatr(Z output_shapeAttributeErrorhasattrr-joinr.Z_inbound_nodesZ_network_nodesrflattenZinbound_layersZget_nodeAssertionErrorrrget_name)'model show_shapes show_dtypeshow_layer_namesr expand_nestedr!r$r rr messagerZsub_n_first_nodeZsub_n_last_nodeZsub_w_first_nodeZsub_w_last_noder7nodeirZlayer_idZ layer_name class_nameZsubmodel_wrapperZ sub_w_nodesZchild_class_nameZsubmodel_not_wrapperZ sub_n_nodesrr)Z outputlabelsZ inputlabelsZnode_keyZ inbound_layerZinbound_layer_idr5Z output_nameZinbound_layer_namer )r,r r@Bs                               r@zkeras.utils.plot_model model.pngc Cst|||||||d}t|}|dkr*dStj|\} } | sDd} n | dd} |j|| d| dkryddlm} | j|d St k rYnXdS) aDConverts a Keras model to dot format and save to a file. Example: ```python input = tf.keras.Input(shape=(100,), dtype='int32', name='input') x = tf.keras.layers.Embedding( output_dim=512, input_dim=10000, input_length=100)(input) x = tf.keras.layers.LSTM(32)(x) x = tf.keras.layers.Dense(64, activation='relu')(x) x = tf.keras.layers.Dense(64, activation='relu')(x) x = tf.keras.layers.Dense(64, activation='relu')(x) output = tf.keras.layers.Dense(1, activation='sigmoid', name='output')(x) model = tf.keras.Model(inputs=[input], outputs=[output]) dot_img_file = '/tmp/model_1.png' tf.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 models into clusters. dpi: Dots per inch. Returns: A Jupyter notebook Image object if Jupyter is installed. This enables in-line display of the model plots in notebooks. )rIrJrKr rLr!NZpng)rAZpdfr)display)filename) r@rospathsplitextwriteZIPythonrSZImager4) rHZto_filerIrJrKr rLr!r_ extensionrSr r r plot_models,,   r[)FFTrFrF)rQFFTrFr)__doc__rUr1Z&tensorflow.python.keras.utils.io_utilsrZtensorflow.python.utilrZ tensorflow.python.util.tf_exportrZpydot_ngrr4Z pydotplusr rrr@r[r r r r sF        M