代码之家  ›  专栏  ›  技术社区  ›  rob me of my freedom

在不使用Y_pred和Y_actual的tensorflow中编写自定义损失

  •  0
  • rob me of my freedom  · 技术社区  · 2 年前

    我的DNN分类任务的损失函数是基于特征值的,不需要输入Y_预测和Y_实际值。是否可以使用Tensorflow编写专门的自定义损耗函数?

    1 回复  |  直到 2 年前
        1
  •  1
  •   YScharf    2 年前

    当然:

    def customLoss(y_true, y_pred, alpha):
                   loss = ....alpha
                    return loss
    model.compile(loss=customLoss(alpha), optimizer='sgd')