wcipriano/pretty-print-confusion-matrix

해결하는 문제

Python에서 MATLAB에서 생성하는 것과 유사한 시각적으로 매력적인 혼동 행렬을 간단하게 생성할 수 있도록 제공하여 분류 모델 성능을 평가하기 쉽게 합니다.

작동 방식

이 라이브러리는 seaborn과 matplotlib을 사용하여 행렬을 렌더링합니다. 미리 계산된 NumPy 행렬(파이썬의 Pandas DataFrame을 통해) 또는 정답 레이블(y_test)과 모델 예측을 포함하는 두 개의 NumPy 배열에서 직접 플롯을 생성할 수 있습니다.

대상 사용자

모델이 데이터를 잘못 분류하는 위치를 분석하기 위해 분류 결과를 시각화해야 하는 데이터 과학자 및 머신러닝 엔지니어.

주요 기능

• DataFrame과 원시 예측 벡터 둘 다에서 플롯 지원.
• 정수 대신 설명적인 텍스트(예: "Dog", "Cat")로 축 레이블을 사용자 정의 가능.
• Matplotlib 컬러맵과 통합되어 행렬의 시각적 색상 스키마를 사용자 정의 가능.