如何解决找到具有张量流的交叉族
predicted_scores = tf.constant([
[0.32,0.2,0.5,0.3,0.5],[0.31,[0.3111,[0.33423,[0.33243,[0.334,[0.32,0.15,0.5]
])# predicted_scores(N,8,n_classes)
true_classes = tf.constant([
[ 5,5,10,5]
])
如果我有这样的预测分数和true_classs
我用过的火炬
conf_loss_all = tf.nn.sigmoid_cross_entropy_with_logits(predicted_scores.view(-1,n_classes),true_classes.view(-1)) # (N * 8732)
找到跨种族的人
如何使用TensorFlow查找交叉熵?
解决方法
您可以使用SparseCategoricalCrossentropy损失。
scce = tf.keras.losses.SparseCategoricalCrossentropy()
scce(true_classes[0],predicted_scores)
<tf.Tensor: shape=(),dtype=float32,numpy=2.8711867>
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。