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 Загрузка pandas DataFrame Трансферное обучение с 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`	Аргумент, передаваемый PyDot, строка, определяющая формат графика: `"TB"` создаёт вертикальный график; `"LR"` создаёт горизонтальный график.
`expand_nested`	Расширять вложенные функциональные модели в кластеры.
`dpi`	Разрешение изображения в точках на дюйм.
`show_layer_activations`	Отображать активации слоёв (только для слоёв, у которых есть свойство `activation`).
`show_trainable`	Выводить, является ли слой обучаемым.

Возвращаемое значение
Объект изображения Jupyter Notebook, если 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