tf.keras.utils.plot_model

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

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.keras.utils.plot_model

tf.keras.utils.plot_model(
    model,
    to_file='model.png',
    show_shapes=False,
    show_dtype=False,
    show_layer_names=True,
    rankdir='TB',
    expand_nested=False,
    dpi=96,
    layer_range=None,
    show_layer_activations=False
)

Пример:

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)

Аргументы
`model`	Экземпляр модели Keras
`to_file`	Имя файла для изображения графика.
`show_shapes`	Отображать ли информацию о форме.
`show_dtype`	Отображать ли типы данных слоев.
`show_layer_names`	Отображать ли имена слоев.
`rankdir`	`rankdir` аргумент, передаваемый в PyDot, строка, определяющая формат графика: 'TB' создает вертикальный график; 'LR' создает горизонтальный график.
`expand_nested`	Расширять ли вложенные модели в кластеры.
`dpi`	Точки на дюйм.
`layer_range`	Вход `list`, содержащий два `str` элемента, которые являются начальным именем слоя и конечным именем слоя (включая оба), указывающими диапазон слоев, для которых будет создан график. Также принимает регулярные выражения вместо точных имён. В этом случае, начальное предикат будет первым элементом, которому он соответствует `layer_range[0]` а конечный предикат будет последним элементом, которому он соответствует `layer_range[1]`. По умолчанию `None` который рассматривает все слои модели. Обратите внимание, что вы должны передать диапазон таким образом, чтобы результирующий подграф был полным.
`show_layer_activations`	Отображать ли активации слоев (только для слоев, у которых есть свойство `activation`).

Возможные исключения
`ValueError`	если `plot_model` вызывается до построения модели.

Возвращаемое значение
Объект изображения 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/versions/r2.9/api_docs/python/tf/keras/utils/plot_model