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