首页 > 解决方案 > 将输入打印到 Keras 模型

问题描述

有什么方法可以将输入打印到我的 Keras 模型中?我不确定“batch_size”是指样本数还是每个样本的数据点数。如果我可以将输入打印到我的 Keras 模型中,那就太好了。

标签: pythonkerasneural-network

解决方案


批量大小是一步通过网络的样本数。因此,将步数乘以批量大小得出样本总数。

batch_size = number_of_samples / number_of_steps

要获取 Keras 模型的输入张量:

input_tensor = model.input

从 获取形状张量input_tensor

shape_tensor = tf.shape( input_tensor )

通过向 a提供一些输入来运行shape_tensora :tf.Session()input_tensor

with tf.Session() as session:
    print( session( shape_tensor , feed_dict={ input_tensor:some_input } ) )

这将打印形状。


推荐阅读