wcipriano/pretty-print-confusion-matrix

解決的問題

提供一種簡單的方法，可在 Python 中建立視覺上吸引人的混淆矩陣，類似於 MATLAB 所產生的結果，使評估分類模型效能更加容易。

工作原理

該程式庫使用 seaborn 和 matplotlib 來渲染矩陣。可從預先計算的 NumPy 矩陣（透過 Pandas DataFrame）或直接從包含真實標籤（y_test）與模型預測的兩個 NumPy 陣列生成圖表。

適用對象

需要可視化分類結果，以更佳分析模型誤分類資料位置的資料科學家與機器學習工程師。

主要特色

• 支援從 DataFrame 與原始預測向量產生圖表。
• 可自訂軸標籤，將整數取代為描述性文字（例如 "Dog", "Cat"）。
• 與 Matplotlib 色圖整合，可自訂矩陣的視覺色彩方案。