首页 > 解决方案 > 在 Keras 中设计自定义损失函数时使用第三个变量列表

问题描述

我正在尝试为体育博彩模型创建损失函数。我无法将赔率(第三个变量)带入损失函数。当我确实将它带入时,它适用于第一批,但似乎在新一批开始时恢复到我的列表/张量的开头。是否可以像 yTrue/yPred 值一样迭代赔率列表。

odds = data_info['FRD_Open'] # Specific value for each model prediction

def custom_Loss(odds):

     def customLoss(yTrue, yPred):

         o = tf.convert_to_tensor(odds, dtype=tf.float32)
         loss = K.mean(K.sum(K.relu((yTrue * (o - 1) - (1 - yTrue)) * yPred), axis=1))

         return -loss

     return customLoss

注意:data_info 是用于训练和测试模型的数据的 pandas 数据框。

标签: pythontensorflowkerasloss-function

解决方案


推荐阅读