diff --git a/matchzoo/losses/rank_cross_entropy_loss.py b/matchzoo/losses/rank_cross_entropy_loss.py index 2be64023..9bf56f43 100644 --- a/matchzoo/losses/rank_cross_entropy_loss.py +++ b/matchzoo/losses/rank_cross_entropy_loss.py @@ -48,7 +48,7 @@ def __call__(self, y_true: np.array, y_pred: np.array) -> np.array: labels.append(neg_labels) logits = K.concatenate(logits, axis=-1) labels = K.concatenate(labels, axis=-1) - return -K.mean(K.sum(labels * K.log(K.softmax(logits)), axis=-1)) + return -K.mean(K.sum(labels * K.log(K.softmax(logits) + np.finfo(float).eps), axis=-1)) @property def num_neg(self):