python - カスタム損失 mse 関数に if 条件を追加する

Question

DNNモデルの小さな値をより強調するために、kerasでカスタム損失mse関数を使用しようとしています。

私はこのようなことを試しました：

import keras.backend as K
def scaled_mse(y_true,y_pred):
    loss = K.square(y_pred - y_true)
    if y_true>0.1:
        loss=loss/K.abs(y_true)
     return loss

私の ML モデルの入力は 41 値です。出力値は 1 つだけです。

しかし、次のエラーが返されます: OperatorNotAllowedInGraphError: Using a tf.Tensoras a Python boolis not allowed in Graph execution. Eager 実行を使用するか、この関数を @tf.function で装飾します。

助けてくれてありがとう！