Keras搭建多输入模型

版权声明：本文为博主原创文章，请尊重原创，转载请注明原文地址和作者信息！ https://blog.csdn.net/zzc15806/article/details/84067017

---

简介

当我们的任务涉及到多个维度不同的数据来拟合一个目标时，我们需要构建多输入模型。

---

模型构建 

假设我们需要搭建如下的模型，输入数据分别为100维和50维的向量，输出为0或1：

from keras.layers import Conv1D, Dense, MaxPool1D, concatenate, Flatten
from keras import Input, Model
from keras.utils import plot_model
import numpy as np


def multi_input_model():
    """构建多输入模型"""
    input1_= Input(shape=(100, 1), name='input1')
    input2_ = Input(shape=(50, 1), name='input2')

    x1 = Conv1D(16, kernel_size=3, strides=1, activation='relu', padding='same')(input1_)
    x1 = MaxPool1D(pool_size=10, strides=10)(x1)

    x2 = Conv1D(16, kernel_size=3, strides=1, activation='relu', padding='same')(input2_)
    x2 = MaxPool1D(pool_size=5, strides=5)(x2)

    x = concatenate([x1, x2])
    x = Flatten()(x)

    x = Dense(10, activation='relu')(x)
    output_ = Dense(1, activation='sigmoid', name='output')(x)

    model = Model(inputs=[input1_, input2_], outputs=[output_])
    model.summary()

    return model

if __name__ == '__main__':
    # 产生训练数据
    x1 = np.random.rand(100, 100, 1)
    x2 = np.random.rand(100, 50, 1)
    # 产生标签
    y = np.random.randint(0, 2, (100,))

    model = multi_input_model()
    # 保存模型图
    plot_model(model, 'Multi_input_model.png')

    model.compile(optimizer='adam', loss='binary_crossentropy')
    model.fit([x1, x2], y, epochs=10, batch_size=10)