分享

Keras模型可视化中的这些(None,x)值是什么?

 袁先森lemon 2021-04-15
rnn模型可视化

model = Sequential()

model.add(Embedding(10000, 300, input_length=200))
model.add(LSTM(256, return_sequences=True, dropout=0.5, recurrent_dropout=0.5))
model.add(LSTM(256, dropout=0.5, recurrent_dropout=0.5))
model.add(Dense(4, activation='softmax'))
...

plot_model(model, to_file='rnn.png' ,show_shapes=True, show_layer_names=True)
为什么是(None,200)而不是(200)?


解决方案
那是由于批量大小。训练模型时,可以通过不同的批次大小(例如32、64,...)。

这意味着,例如,如果您训练批处理大小为32的模型,则第一层的形状将为(32,200),依此类推。

构建模型时,仍未定义输入批处理大小。这就是为什么Tensorflow进行打印的原因None。

    本站是提供个人知识管理的网络存储空间,所有内容均由用户发布,不代表本站观点。请注意甄别内容中的联系方式、诱导购买等信息,谨防诈骗。如发现有害或侵权内容,请点击一键举报。
    转藏 分享 献花(0

    0条评论

    发表

    请遵守用户 评论公约

    类似文章 更多