首页 > 解决方案 > 在 google colab 的 keras 中使用连接时出错

问题描述

the_error_i_received

图片中提到了错误。

TypeError                                 Traceback (most recent call last)
<ipython-input-200-8ea67f851d87> in <module>()
      1 final_model = Sequential([
----> 2   Concatenate([image_model, caption_model], mode='concat', concat_axis=1),
      3   Bidirectional(LSTM(256, return_sequences=False)),
      4   Dense(vocab_size),
      5   Activation('softmax')

标签: kerasconcatenationgoogle-colaboratorykeras-layer

解决方案


您错误地使用了 Concatenate 层。将您的代码更改为以下内容:

final_model = Sequential([
             Concatenate([image_model, caption_model], axis=1),
             Bidirectional(LSTM(256, return_sequences=False))
             Dense(vocab_size), 5 Activation('softmax'),
             Activation('softmax')
])

推荐阅读