Keras plot_model将模型体系结构中的问号替换为None

提问者：小点点

我想获得我正在工作的lstm模型的模型架构。我知道问号表示批处理大小，这是没有指定的，所以它显示问号，但我希望它被替换为无。您能不能帮我如何用None替换问号，因为我大部分时候都看到plot_model显示的批处理大小为None？下面是我使用的代码和我得到的图像。

    tf.keras.utils.plot_model(
    model,
    to_file='model.png',
    show_shapes=True,
    show_layer_names=True
    )

我得到的模型架构图像

共1个答案

匿名用户

转到第三方软件包所在的文件夹，转到

tensorflow/keras/utils/vis_utils.py

并更改行208-209:

def format_shape(shape):
    return str(shape).replace(str(None), '?')

用“无”更改“？”

您可以找到如下源代码:

import tensorflow as tf

print(tf.__file__)

'C:\Users\User\anaconda3\envs\tf\lib\site-packages\tensorflow\__init__.py'

然后可以找到正确的.py文件。