搜索多个输出损失函数(tensorflow + keras)

时间:2019-12-02 13:59:34

标签: tensorflow keras loss-function multipleoutputs generative

我尝试使用NN作为生成器,但不知道应该采用哪种损失函数。

它输入噪声,并应输出0到1之间的3个数字。这些数字已由我要用来训练生成模型的另一个模型(好坏)进行分类。

model = keras.Sequential([
    keras.layers.Dense(10, activation='relu', input_shape = [1]),
    keras.layers.Dense(10, activation='relu'),
    keras.layers.Dense(3, activation='tanh')])

model.compile(optimizer='adam',
              loss='???',
              metrics=['accuracy'])

如果我正确理解了问题,则需要一个损失函数,该函数输入一个目标和一个预测以优化多个输出。

任何人都知道该怎么办?我需要更改模型吗?采取哪种激活功能?对不起,我只能找到图像生成的例子,没什么。

0 个答案:

没有答案