最高得点のクラスを取得する

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