如何指定应在Keras中计算哪个输出指标

时间:2019-10-21 04:53:58

标签: keras tf.keras

我想从Keras模型返回两个值。一个是浮点数,另一个是字符串:

output = tf.keras.layers.Dense(1, activation='linear', name='...')(h2)
input_key = tf.keras.layers.Lambda(lambda x: x, name='...')(inputs['key'])

当我使用以下方法创建模型时:

model = tf.keras.models.Model(inputs, [output, input_key])
model.compile(optimizer='adam', loss='mse', metrics=['mse'])

我收到此错误:

 Value passed to parameter 'x' has DataType string not in list of allowed values: bfloat16, float16, float32, float64, int32, int64, complex64, complex128

因为损失,指标等假设只有一个输出,而第二个输出是字符串。

最简单的解决方法是什么?

0 个答案:

没有答案