TensorFlow tf.keras.losses.SparseCategoricalCrossentropy
tensorflow
一个面向所有人的开源机器学习框架
项目地址:https://gitcode.com/gh_mirrors/te/tensorflow
免费下载资源
·
下面的例子,y_true
的形状是[batch_size]
,y_pred
的形状是[batch_size,num_class]
.
也就是说y_pred
经过softmax
层之后,是one-hot
编码,SparseCategoricalCrossentropy
将y_true
变成了one-hot
编码.
cce = tf.keras.losses.SparseCategoricalCrossentropy()
loss = cce(
[0, 1, 2],
[[.9, .05, .05], [.5, .89, .6], [.05, .01, .94]])
print('Loss: ', loss.numpy()) # Loss: 0.3239
init
__init__(
from_logits=False,
reduction=losses_utils.ReductionV2.AUTO,
name=None
)
call
__call__(
y_true,
y_pred,
sample_weight=None
)
参考:
官网
GitHub 加速计划 / te / tensorflow
184.54 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:26 天前 )
a49e66f2
PiperOrigin-RevId: 663726708
1 个月前
91dac11a
This test overrides disabled_backends, dropping the default
value in the process.
PiperOrigin-RevId: 663711155
1 个月前
更多推荐
已为社区贡献18条内容
所有评论(0)