获得最高分的班级

Question

我使用以下代码来获得预测。

for node_id in top_k:
    score = predictions[0][node_id]
    human_string = label_lines[node_id]
    print('%s (score = %.5f)' % (human_string, score))

上面的代码打印了所有类的分数预测。但是，如何获得得分最高的标签？

Answer 1

这是使用max的解决方案

score, key = max((v, k) for k, v in predictions[0].items())
print('{} (score = {:.5f})'.format(label_lines[key], score))