代码之家  ›  专栏  ›  技术社区  ›  Sreeram TP

Keras中不同数据类型的深度学习模型

  •  4
  • Sreeram TP  · 技术社区  · 7 年前

    我试图用Keras建立一个分类模型。我的数据包含一些数字特征和一些文本特征。我所说的文字特征是指评论或类似的东西。数字特征将是类别、年龄等。

    我想将文本功能传递到嵌入层,然后传递到LSTM层。数字特征需要传递到一系列密集层。之后,需要连接两个层。之后,形成致密层以进行输出。

    我如何在Keras中实现这种类型的模型。?

    或者是否有其他方法可以同时在模型中使用数字特征和基于文本的特征。?

    1 回复  |  直到 7 年前
        1
  •  3
  •   S.Mohsen sh    7 年前

    使用keras函数API实现这样的网络相当容易。 假设已定义两个连续模型来处理文本和数字特征,则可以合并输出并继续使用更多层:

    txt_input = keras.layers.Input(shape=(n,))
    txt_feat = text_network(txt_input)
    num_input = keras.layers.Input(shape=(m,))
    num_feat = num_network(input2)
    concatinated = keras.layers.Concatenate()([txt_feat, num_feat])
    
    out = keras.layers.Dense(nodes)(concatinated)
    model = keras.models.Model(inputs=[input1, input2], outputs=out)
    

    您还可以使用其他类型的合并 merge layer 来自keras。