为一个输入训练具有两个可能正确输出的神经网络
数据挖掘
机器学习
神经网络
数据集
卷积神经网络
训练
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