keras - 单个指标返回多个值

Question

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

我能做的是将相同的指标调用 6 次，但我认为它会独立计算 6 次，我真的很想避免这种情况。keras 有没有办法返回数组的所有 6 个值，即使是nx 6输出也比这种行为更受欢迎。

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

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

    return model_quality

我想在train和predict阶段分别计算这些指标。

标签： kerastensorflow2.0