Keras-多输入多输出实例(多任务)

Keras-多输入多输出实例多输入多输出实例(多任务多任务)

主要介绍了Keras-多输入多输出实例(多任务)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来

看看吧

1、模型结果设计

2、代码

from keras import Input, Model

from keras.layers import Dense, Concatenate

import numpy as np

x1 = rd.rand(samples_n, samples_dim_01)

x2 = rd.rand(samples_n, samples_dim_02)

y_1 = []

y_2 = []

y_3 = []

for x11, x22 in zip(x1, x2):

y_1.append(np.sum(x11) + np.sum(x22))

y_2.append(np.max([np.max(x11), np.max(x22)]))

y_3.append(np.min([np.min(x11), np.min(x22)]))

y_1 = np.array(y_1)

inputs_01 = Input((samples_dim_01,), name='input_1')

inputs_02 = Input((samples_dim_02,), name='input_2')

# 全连接层

dense_01 = Dense(units=3, name="dense_01", activation='softmax')(inputs_01)

dense_011 = Dense(units=3, name="dense_011", activation='softmax')(dense_01)

dense_02 = Dense(units=6, name="dense_02", activation='softmax')(inputs_02)

# 加入合并层

merge = Concatenate()([dense_011, dense_02])