向 keras 中的内置损失函数添加新术语

时间:2021-01-27 15:21:07

标签: python tensorflow keras optimization loss-function

我正在使用以下损失函数来训练我正在使用的神经网络:

loss = {'time_prediction': keras.losses.mean_squared_error,
            'decoded_mean': keras.losses.mean_squared_error}

现在,我需要通过向先前的损失函数添加一个新术语来更新我的损失函数,这不是 keras 中的内置术语。如何将这个新的自定义函数添加到之前的损失函数中? 我将新术语写成如下函数:

def new_term_loss(X_alpha, X_train):
  return (1/np.size(X_alpha)) * np.sum((X_alpha - X_train)>0)

0 个答案:

没有答案
相关问题