Keras文档:多输入和多输出模型无法遵循

时间:2018-03-08 15:38:03

标签: python keras

我正在关注页面上的示例:Multi-input and multi-output models

模型设置用于预测转发和喜欢新闻标题的数量。那么 main_output 是预测转发的数量和 aux_output 预测喜欢的数量?

Button

我在AttributeError中遇到错误:'list'对象没有属性'ndim'

1 个答案:

答案 0 :(得分:1)

您的输入/输出必须是NumPy数组,其中第一个维度是批量大小。例如:

headline_data = np.random.randint(1, 10000 + 1, size=(32, 100))
additional_data = np.random.randint(1, 10000 + 1, size=(32, 5))
labels = np.random.randint(0, 1 + 1, size=(32, 1))

请注意,这是一个玩具示例,我们正在随机生成输入。