tf.keras.utils.plot_model
Преобразует модель Keras в формат DOT и сохраняет её в файл.
tf.keras.utils.plot_model(
model, to_file='model.png', show_shapes=False, show_layer_names=True,
rankdir='TB', expand_nested=False, dpi=96
)
| Аргументы |
model | Экземпляр модели Keras |
to_file | Имя файла изображения графика. |
show_shapes | Показывать ли информацию о форме. |
show_layer_names | Показывать ли имена слоев. |
rankdir | Аргумент, передаваемый в PyDot, строка, определяющая формат графика: 'TB' создаёт вертикальный график; 'LR' создаёт горизонтальный график. |
expand_nested | Развернуть вложенные модели в кластеры. |
dpi | Точки на дюйм. |
| Возвращаемое значение |
| Объект изображения Jupyter Notebook, если Jupyter установлен. Это позволяет отображать графики моделей внутри ноутбуков. |