首页 > 解决方案 > Tensorflow / Keras 自定义损失函数

问题描述

我目前正在尝试在 Keras 中编写自己的损失函数,它检查我的预测值是否以任何顺序存在于标签中。

这是一个用python编写的代码示例:

def my_loss(y_true, y_pred):
    n_values = 5
    loss = 0
    for i in range(n_values):
        if y_pred[i] not in y_true:
            loss += 1
    return loss

我不知道如何用keras.backend. 我什至无法找到 kerasbackend.sum(..), .flatten等函数的文档。

标签: pythontensorflowkeras

解决方案


推荐阅读