How to calculate top5 accuracy?

2019-03-11  本文已影响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))
上一篇 下一篇

猜你喜欢

热点阅读