0

我有一个创建如下的模型:

tf.keras.Model(some_x, [some_y1, some_y2])

它编译如下:

model.compile(
    optimizer="Adam",
    loss=[
        loss_function(somefoo1),
        loss_function(somefoo2),
    ],
    loss_weights=[0.5, 0.5],
    metrics=[
        [
            accuracy_function(somefoo1, "acc1")
        ],
        [
            accuracy_function(somefoo2, "acc2")
        ],
    ],
)

这是准确度函数的样子:

def accuracy_function(somefoo, name):
    def calculate(y_true, y_pred):
         #logic
    calculate.__name__ = name
    return calculate

问题是 model.metric_names 给出的名称类似于 [conv2d_9_acc1', 'conv2d_12_acc2'] 我怎样才能摆脱它前面的图层名称?我可以命名我的输出层吗?那应该可以解决问题

4

0 回答 0