Make sure output sum equals 1

keras-team · Mar 9, 2023 · d3dd32f · d3dd32f
1 parent 40e547f
commit d3dd32f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/keras/backend.py b/keras/backend.py
@@ -5585,7 +5585,7 @@ def categorical_focal_crossentropy(
  from_logits=False,
  axis=-1,
 ):
- 
+
  output, from_logits = _get_logits(
  output, from_logits, "Softmax", "categorical_focal_crossentropy"
  )
@@ -5596,6 +5596,8 @@ def categorical_focal_crossentropy(
  lambda: output,
  )
 
+ output = output / tf.reduce_sum(output, axis=axis, keepdims=True)
+
  epsilon_ = _constant_to_tensor(epsilon(), output.dtype.base_dtype)
  output = tf.clip_by_value(output, epsilon_, 1.0 - epsilon_)