为一个输入训练具有两个可能正确输出的神经网络

数据挖掘 机器学习 神经网络 数据集 卷积神经网络 训练
2022-03-07 12:49:58

我有一个系统作为黑盒,对于单个输入样本有两个正确的输出。现在我想训练一个神经网络为该输入样本生成至少一个正确的输出。我该怎么办🤷🏻‍♂️

数据集

1个回答

使用 Keras,您可以使用功能 API 来估计具有两个输出的模型(“多输出”)。只需在两个输出上训练模型,例如:

# Outputs
out1 = Dense(1)(x)
out2 = Dense(1)(x)

# Compile/fit the model
model = Model(inputs=Input_1, outputs=[out1,out2])
model.compile(optimizer = "rmsprop", loss = 'mse')

# Add actual data here in the fit statement
model.fit(train_data, [train_targets,train_targets2], epochs=500, batch_size=4, verbose=0, validation_split=0.2)

在这里找到完整的模型:https ://github.com/Bixi81/Python-ml/blob/master/keras_multitarget_functionalAPI.py