使用Keras的同一神经网络进行分类和回归

Question

我想构建一个神经网络，同时输出一个分类标签和一个回归值。我想使用Keras来做到这一点。现在我的代码仅用于分类：

 mdl = Sequential()
 mdl.add(Dense(100, activation='relu', input_dim=X_train.shape[1]))
 mdl.add(Dense(200, activation='relu'))
 mdl.add(Dense(100, activation='relu'))
 mdl.add(Dense(6, activation='softmax'))

 mdl.compile(loss='categorical_crossentropy', optimizer='sgd', metrics=['accuracy'])

 # early stopping implementation
 filepath="weights.best.hdf5"
 checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, 
 save_best_only=True, mode='max')
 early_stop = EarlyStopping(monitor='val_acc', patience=100, mode='max') 
 callbacks_list = [checkpoint, early_stop]


 # fit network
 history = mdl.fit(X_train, y_train, epochs=2000, batch_size=32, 
 validation_split=0.2, verbose=2, shuffle=True, callbacks=callbacks_list)

现在我在输出层上有一个softmax激活函数，对应于我用于分类的概率。如何修改此代码以同时输出表示回归问题的continuos值。我知道Keras函数API允许指定多输入和多输出网络。谁知道我该怎么做？

Answer 1

您的代码可以直接转换为Keras函数API，如他们的文档所示。您需要更改顺序声明

mdl = Sequential()
mdl.add(Dense(100, activation='relu', input_dim=X_train.shape[1]))
mdl.add(Dense(200, activation='relu'))
mdl.add(Dense(100, activation='relu'))
mdl.add(Dense(6, activation='softmax'))

与其功能等效：

inputs = Input(shape=(X_train.shape[1],))
layer1 = Dense(100, activation='relu')(inputs)
layer2 = Dense(200, activation='relu')(layer1)
layer3 = Dense(100, activation='relu')(layer2)
classifier = Dense(6, activation='softmax')(layer3)
mdl = Model(inputs=inputs, outputs=classifier)

（通常人们只是对所有中间层重复使用相同的变量，甚至在文档示例中也会这样做，但这个IMHO更清楚一些）。

完成后，可以添加另一个输出层，该输出层从最后一个密集层3“分支”，并设置模型有两个输出，例如：

regression = Dense(1, activation='linear')(layer3)
mdl = Model(inputs=inputs, outputs=[classifier, regression])