sklearn.metrics.confusion_matrix

目的:評估分類器準確性
函數:sklearn.metrics.confusion_matrix(y_true, y_pred, labels=None, sample_weight=None)
輸入:

  • y_true:實際的目標結果
  • y_pred:預測的結果
  • labels: 標簽,對結果中的string進行排序, 順序對應0、1、2
  • sample_weight:樣本的權重?

輸出:

  • 一個矩陣,shape=[y中的類型數,y中的類型數]
  • 矩陣中每個值表征分類的準確性
  • 第0行第0列的數表示y_true中值為0,y_pred中值也為0的個數
  • 第0行第1列的數表示y_true中值為0,y_pred中值為1的個數

示例:

>>> from sklearn.metrics import confusion_matrix
>>> y_true = [2, 0, 2, 2, 0, 1] 
>>> y_pred = [0, 0, 2, 2, 0, 2]
>>> confusion_matrix(y_true, y_pred)
 array([[2, 0, 0], [0, 0, 1], [1, 0, 2]])
>>> y_true = ["cat", "ant", "cat", "cat", "ant", "bird"] 
>>> y_pred = ["ant", "ant", "cat", "cat", "ant", "cat"]
>>> confusion_matrix(y_true, y_pred, labels=["ant", "bird", "cat"]) 
array([[2, 0, 0], [0, 0, 1], [1, 0, 2]])
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容