We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7255c62 commit 60e82efCopy full SHA for 60e82ef
tensorflow_addons/losses/focal_loss.py
@@ -41,7 +41,7 @@ class SigmoidFocalCrossEntropy(LossFunctionWrapper):
41
>>> fl = tfa.losses.SigmoidFocalCrossEntropy()
42
>>> loss = fl([[0.97], [0.91], [0.03]], [[1.0], [1.0], [0.0]])
43
>>> loss
44
- <tf.Tensor: shape=(3,), dtype=float32, numpy=array([0.00010971, 0.00329749, 0.00030611], dtype=float32)>
+ <tf.Tensor: shape=(3,), dtype=float32, numpy=array([10971, 0.00329749, 0.00030611], dtype=float32)>
45
46
Usage with tf.keras API:
47
0 commit comments