tf.keras.utils.model_to_dot
Преобразуйте Keras модель в формат dot.
tf.keras.utils.model_to_dot(
model, show_shapes=False, show_dtype=False, show_layer_names=True,
rankdir='TB', expand_nested=False, dpi=96, subgraph=False
)
| Аргументы |
model | Экземпляр Keras модели. |
show_shapes | отображать ли информацию о форме. |
show_dtype | отображать ли типы данных слоев. |
show_layer_names | отображать ли имена слоев. |
rankdir | rankdir аргумент, переданный PyDot, строка, определяющая формат графика: 'TB' создает вертикальный график; 'LR' создает горизонтальный график. |
expand_nested | расширять ли вложенные модели в кластеры. |
dpi | Точки на дюйм. |
subgraph | возвращать ли экземпляр pydot.Cluster. |
| Возвращает |
Экземпляр pydot.Dot , представляющий Keras модель, или экземпляр pydot.Cluster , представляющий вложенную модель, если subgraph=True. |
| Возможные исключения |
ImportError | если graphviz или pydot недоступны. |