多分類問題下的評估指標

一般二分類中直接用 查準率(precision)、查全率(recall)、F1 score 、AUC就可以做評價指標

在多分類中:

1、西瓜書中介紹了 宏查準率(macro-P)、宏查全率(macro-R)、宏F1(macro-F1) 、以及 微查準率(micro-P)、微查全率(micro-R)、微F1(micro-F1)。可自行查閱

在tensorflow中,可以使用以下任意函數(兩者等價)獲得混淆矩陣,從而獲得上述分類的評價指標

  • tf.confusion_matrix
  • tf.math.confusion_matrix

 

函數介紹具體參見:https://tensorflow.google.cn/api_docs/python/tf/math/confusion_matrix

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章