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=200,
subgraph=False,
show_layer_activations=False,
show_trainable=False,
**kwargs
)
| Аргументы | |
|---|---|
model | Экземпляр модели Keras. |
show_shapes | отображать ли информацию о формах. |
show_dtype | отображать ли типы данных слоёв. |
show_layer_names | отображать ли названия слоёв. |
rankdir | rankdir аргумент, передаваемый в PyDot, — строка, задающая формат графика: "TB" создаёт вертикальный график; "LR" создаёт горизонтальный график. |
expand_nested | разворачивать ли вложенные модели Functional в кластеры. |
dpi | Разрешение изображения в точках на дюйм. |
subgraph | возвращать ли экземпляр pydot.Cluster. |
show_layer_activations | Отображать ли функции активации слоёв (только для слоёв со свойством activation). |
show_trainable | отображать ли, является ли слой обучаемым. |
| Возвращает | |
|---|---|
Экземпляр pydot.Dot, представляющий модель Keras, или экземпляр pydot.Cluster, представляющий вложенную модель, если задано subgraph=True. |
© 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/model_to_dot