Keras:将模型的结果连接到其他特征

我正在使用 Keras 训练模型,该模型将文本和其他特征作为输入。

我打算执行以下操作:

  1. 用文本训练一个 LSTM 模型,输出一个 32 维的结果。

  2. 将 LSTM 模型(32 维)的输出与其他特征(15 维)连接起来,生成一个 47 维的输入。

  3. 使用这个 47 维输入训练另一个模型。

我如何在 Keras 中执行此 Concat 操作?有任何想法吗?


慕无忌1623718
浏览 90回答 1
1回答

冉冉说

使用tf.keras.layers.concatenate(     inputs, axis=-1, **kwargs )这是官方文档提供的示例不过,需要注意一些关键事项:选择正确的损失函数,它可能是一些单独损失的加权和是否值得联合或单独训练网络
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python