获得最高分的课程

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))