gpt4 book ai didi

neural-network - 如何在 tensorflow 中实现多类铰链损失

转载 作者:行者123 更新时间:2023-12-04 04:25:54 29 4
gpt4 key购买 nike

我想在 tensorflow 中实现多类铰链损失。配方如下:

multi-class hinge loss function

当预测正确时,我发现很难获得第二个最大预测概率。我试过用tf.nn.top_k来计算,可惜tf.nn.top_k没有实现梯度操作。那么我该如何实现呢?

最佳答案

top_k有渐变,在 0.8 版中添加 here

关于neural-network - 如何在 tensorflow 中实现多类铰链损失,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36904298/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com