更加严格

ypw757 · May 7, 2022 · b465d50 · b465d50
1 parent 9271dcf
commit b465d50
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/bert4keras/backend.py b/bert4keras/backend.py
@@ -343,8 +343,9 @@ def multilabel_categorical_crossentropy(y_true, y_pred):
     n_mask = K.less_equal(y_true, 1 - K.epsilon())
     p_mask = K.greater_equal(y_true, K.epsilon())
     y_true = K.clip(y_true, K.epsilon(), 1 - K.epsilon())
-    y_neg = K.switch(n_mask, y_pred, -y_pred) + K.log(1 - y_true)
-    y_pos = K.switch(p_mask, -y_pred, y_pred) + K.log(y_true)
+    infs = K.zeros_like(y_pred) + K.infinity()
+    y_neg = K.switch(n_mask, y_pred, -infs) + K.log(1 - y_true)
+    y_pos = K.switch(p_mask, -y_pred, -infs) + K.log(y_true)
     zeros = K.zeros_like(y_pred[..., :1])
     y_neg = K.concatenate([y_neg, zeros], axis=-1)
     y_pos = K.concatenate([y_pos, zeros], axis=-1)