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 | раскрывать вложенные функциональные модели в кластеры. |
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