wcipriano/pretty-print-confusion-matrix

何を解決するか

Pythonで視覚的に魅力的な混同行列を作成する簡単な方法を提供します。MATLABで生成されるものと似ており、分類モデルのパフォーマンスを評価しやすくなります。

動作方法

このライブラリは seaborn と matplotlib を使用して行列をレンダリングします。事前に計算された NumPy 行列（Pandas DataFrame を通じて）または、正解ラベル（y_test）とモデル予測を含む2つの NumPy 配列から直接プロットを生成できます。

対象ユーザー

モデルがデータを誤分類している場所をより良く分析するために分類結果を可視化する必要があるデータサイエンティストや機械学習エンジニア。

特徴

• DataFrame と生の予測ベクトルの両方からプロットをサポート。
• 整数ではなく説明的なテキスト（例："Dog", "Cat"）で軸ラベルをカスタマイズ可能。
• Matplotlib のカラーマップと統合され、行列の視覚的カラースキームをカスタマイズ可能。