Keras 功能 API 是您可以解决问题的一种方式。使用 keras 函数式 API,我们可以构建更像图的模型,如下所示:

为了构建这样的模型,您可以使用 keras,如下所示:
from keras.models import Model
from keras import layers
from keras import Input
input_layer = Input(shape=(100,), dtype='float32', name="Input")
split_layer = layers.Dense(32, activation='relu', name='split_layer')(input_layer)
first_layer = layers.Dense(32, activation='relu', name='first_layer')(split_layer)
second_layer = layers.Dense(32, activation='relu', name='second_layer')(split_layer)
model = Model(input_layer,[first_layer, second_layer])
model.summary()
为了编译这个模型,我们可以为不同的层定义不同的损失函数
model.comile(optimizer=optimizer,
loss={'first_layer':'mse', 'second_layer':'binary_crossentropy'},
metrics=['accuracy'])
完成网络构建后,您可以简单地按如下方式拟合数据:
model.fit(X,
{'first_layer': first_layer_Y,
'second_layer': second_layer_targets},
epochs=10
)