Skip to content

Commit 60e82ef

Browse files
This should fail if the docstrings are tested.
1 parent 7255c62 commit 60e82ef

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_addons/losses/focal_loss.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ class SigmoidFocalCrossEntropy(LossFunctionWrapper):
4141
>>> fl = tfa.losses.SigmoidFocalCrossEntropy()
4242
>>> loss = fl([[0.97], [0.91], [0.03]], [[1.0], [1.0], [0.0]])
4343
>>> loss
44-
<tf.Tensor: shape=(3,), dtype=float32, numpy=array([0.00010971, 0.00329749, 0.00030611], dtype=float32)>
44+
<tf.Tensor: shape=(3,), dtype=float32, numpy=array([10971, 0.00329749, 0.00030611], dtype=float32)>
4545
4646
Usage with tf.keras API:
4747

0 commit comments

Comments
 (0)