獲得最高分的課程

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