美文网首页
How to calculate top5 accuracy?

How to calculate top5 accuracy?

作者: 夕宝爸爸 | 来源:发表于2019-03-11 22:24 被阅读0次

https://stackoverflow.com/questions/42327006/how-to-calculate-top5-accuracy-in-keras/42328189#42328189

top1 = 0.0
top5 = 0.0    
class_probs = model.predict(x)
for i, l in enumerate(labels):
    class_prob = class_probs[i]
    top_values = (-class_prob).argsort()[:5]
    if top_values[0] == l:
        top1 += 1.0
    if np.isin(np.array([l]), top_values):
        top5 += 1.0

print("top1 acc", top1/len(labels))
print("top1 acc", top5/len(labels))

相关文章

网友评论

      本文标题:How to calculate top5 accuracy?

      本文链接:https://www.haomeiwen.com/subject/omoipqtx.html