tf.keras.utils.plot_model

Преобразует модель Keras в формат dot и сохраняет в файл.

tf.keras.utils.plot_model(
    model,
    to_file='model.png',
    show_shapes=False,
    show_dtype=False,
    show_layer_names=False,
    rankdir='TB',
    expand_nested=False,
    dpi=200,
    show_layer_activations=False,
    show_trainable=False,
    **kwargs
)

Используется в блокнотах

Используется в руководствах
Загрузка текста pix2pix: преобразование изображений в изображения с помощью условной GAN Загрузка DataFrame pandas Перенос обучения с YAMNet для классификации звуков окружающей среды Сегментация изображений

Пример:

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)

Аргументы
`model`	Экземпляр модели Keras
`to_file`	Имя файла изображения графика.
`show_shapes`	нужно ли отображать информацию о форме.
`show_dtype`	нужно ли отображать типы данных слоёв.
`show_layer_names`	нужно ли отображать имена слоёв.
`rankdir`	аргумент `rankdir`, передаваемый в PyDot: строка, указывающая формат графика: `"TB"` создаёт вертикальный график; `"LR"` создаёт горизонтальный график.
`expand_nested`	нужно ли разворачивать вложенные функциональные модели в кластеры.
`dpi`	Разрешение изображения в точках на дюйм.
`show_layer_activations`	Отображать активации слоёв (только для слоёв со свойством `activation`).
`show_trainable`	нужно ли отображать, является ли слой обучаемым.

Возвращает
Объект Image для блокнота Jupyter, если установлен Jupyter. Это позволяет отображать графики моделей непосредственно в блокнотах.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/utils/plot_model