Keras 返回多个值的单个度量

Keras 返回多个值的单个度量,keras,tensorflow2.0,Keras,Tensorflow2.0,我最近从tensorflow v1 api升级到tensorflow v2 api,似乎被不希望的行为难住了。我有一个指标,它返回一个在tensorflow v1中有效的数组。然而,keras决定将我的度量合并成一个单一的值,这是不受欢迎的行为 def iou_loss(self): def model_quality(p, y): # brevity return [true_positive, false_positive, false_negati

我最近从tensorflow v1 api升级到tensorflow v2 api,似乎被不希望的行为难住了。我有一个指标,它返回一个在tensorflow v1中有效的数组。然而,keras决定将我的度量合并成一个单一的值,这是不受欢迎的行为

def iou_loss(self):
    def model_quality(p, y):
        # brevity

        return [true_positive, false_positive, false_negative, completeness, correctness, quality]

    return model_quality
我能做的就是调用相同的度量6次,但我会假设它会独立计算6次,这是我真正想要避免的。keras是否有办法返回阵列的所有6个值,甚至n x 6输出也比此行为更可取

def iou_loss(self):
    def model_quality(p, y):
        # brevity

        return [true_positive, false_positive, false_negative, completeness, correctness, quality]

    return model_quality
我想在
训练
预测
阶段分别计算这些指标