我想从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
因为损失,指标等假设只有一个输出,而第二个输出是字符串。
最简单的解决方法是什么?