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



我的DNN分类任务的损失函数是基于特征值的,不需要输入Y_prediction和Y_actual。是否有可能编写专门的自定义损失函数,如使用Tensorflow?

当然:

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

相关内容

  • 没有找到相关文章

最新更新