我的 DNN 分類任務的損失函式是基于特征值的,不需要輸入 Y_prediction 和 Y_actual。是否可以使用 Tensorflow 撰寫專門的自定義損失函式?
uj5u.com熱心網友回復:
當然:
def customLoss(y_true, y_pred, alpha):
loss = ....alpha
return loss
model.compile(loss=customLoss(alpha), optimizer='sgd')
轉載請註明出處,本文鏈接:https://www.uj5u.com/yidong/475732.html
上一篇:如何使用經過訓練的模型預測MNIST資料(輸入形狀的預期軸-1的值為784,但接收到的輸入形狀為(784,1))
