用模型预测最终生成的文件问题

[BucherLi]

我看模型最后生成的.tsv格式结果文件都是概率值，怎么判断哪列对应哪个分类呢

[liuyunwww]

在哪个文件看得到概率值？

[BucherLi]

在哪个文件看得到概率值？

test_results.tsv，前提是在运行python3 run_classifier.py 时加上--do_predict参数

[midneet]

I also have the same questions, how do I tell the label that corresponding to the highest probability?

[midneet]

我後來想出了一個比較麻煩的方法，用比較少的數據集 同時做在do_eval and do_predict上，這樣能同時得到概率值以及accuracy，挑出一個可以達到1.0 accuracy的數據集，然後再從概率值陣列中取argmax這樣兩邊label 與argmax index可以一一對應找出對應的分類