在定义新的损失函数时，为keras中的张量赋值

Question

我正在尝试在Keras中定义一个损失函数

def rmseApprox(y_true, y_pred):
    dum = y_pred
    dum[y_pred>=0]=1.1
    dum[y_pred<0]=1

    return k.abs(K.mean(y_true - dum*y_pred), axis=-1)

将正值增加1.1倍，并将其与真实值进行比较。我收到以下错误：

TypeError: 'Tensor' object does not support item assignment

Answer 1

损失函数是张量和计算图的一部分。因此必须使用https://firebase.google.com/docs/functions/beta-v1-diff?authuser=0#new_properties_for_user_auth_information定义它，并且不像“常规”numpy数组那样。

此示例适用于您：

1.05

请注意，在上述代码中，案例{{1}}的权重为{{1}}。