wcipriano/pretty-print-confusion-matrix

Confusion Matrix in Python: plot a pretty confusion matrix (like Matlab) in python using seaborn and matplotlib

What it solves

It provides a simple way to create visually appealing confusion matrices in Python, similar to those produced by MATLAB, making it easier to evaluate classification model performance.

How it works

The library uses seaborn and matplotlib to render the matrix. It can generate a plot from either a pre-computed NumPy matrix (via a Pandas DataFrame) or directly from two NumPy arrays containing the ground truth labels (y_test) and the model predictions.

Who it’s for

Data scientists and machine learning engineers who need to visualize classification results for better analysis of where a model is misclassifying data.

Highlights

  • Support for plotting from both DataFrames and raw prediction vectors.
  • Customizable axis labels to replace integers with descriptive text (e.g., "Dog", "Cat").
  • Integration with Matplotlib colormaps to customize the visual color scheme of the matrix.

Related

  • Project
    sepandhaghighi/pycm

    A multi-class confusion matrix library for Python that provides extensive statistical metrics and visualization tools for evaluating classification models.

  • Project
    skrub-data/skrub

    skrub is a Python library that simplifies the process of using dataframes for machine learning workflows.

  • Project
    databrickslabs/dbldatagen

    A Python library for generating scalable synthetic data within Databricks using Spark, designed for testing, benchmarking, and demos.

  • Project
    microsoft/semantic-link-labs

    A Python library for Microsoft Fabric notebooks that automates the management, migration, and optimization of Power BI semantic models, reports, and lakehouses.

  • Project
    PAIR-code/lit

    A visual, interactive ML model-understanding tool that supports text, image, and tabular data to help developers debug and interpret model predictions.