比较logits和labels的最佳方法是什么?

2024-03-28 14:52:14 发布

您现在位置:Python中文网/ 问答频道 /正文

我将logits与循环中的标签进行比较:

  for r in range(logits.shape[0]):
    if labels[r] == np.argmax(logits[r]):
      guessed += 1.0

其中labels是整数标签的一维数组,logits是二维数组,第二维是标签的概率。你知道吗

上面的解决方案是一个Python循环,效率不高。应该有一个常用的numpytensorflow快捷方式来做到这一点。你能推荐一个吗?你知道吗


Tags: inforlabelsifnprange整数标签