Plot a pretty confusion matrix (like Matlab) in python using seaborn and matplotlib
Project description
ConfPlot: Plot Confusion Matrix in Python
Plot a pretty confusion matrix (like Matlab) in python using seaborn and matplotlib
This module lets you plot a pretty looking confusion matrix from a np matrix or from a prediction results and actual labels.
Sample plots:
Getting started
Installation
pip install confplot
Usage
Plot confusion matrix from matrix
# import package
import confplot
# assume you have a confusion matrix array like this
array = np.array(
[[13, 0, 1, 0, 2, 0],
[ 0, 50, 2, 0, 10, 0],
[ 0, 13, 16, 0, 0, 3],
[ 0, 0, 0, 13, 1, 0],
[ 0, 40, 0, 1, 15, 0],
[ 0, 0, 0, 0, 0, 20]]
)
# convert it to a pandas dataframe
df_cm = DataFrame(array, index=range(1, 7), columns=range(1, 7))
# create and save confusion matrix plot as "cm_plot.png"
confplot.plot_confusion_matrix_from_matrix(df_cm, outfile="cm_plot.png")
Plot confusion matrix from data
# import package
import confplot
# assume you have 1D y_true (actual values) and y_pred (predictions) arrays
y_true = ...
y_pred = ...
# arange targetclass names if you want
columns = ["ahududu", "ananas", "armut", "avokado", "ayva"]
# create and save confusion matrix plot as "cm_plot.png"
confplot.plot_confusion_matrix_from_data(
y_true,
y_pred,
columns,
outfile="cm_plot.png"
)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
confplot-0.1.1.tar.gz
(6.3 kB
view hashes)
Built Distribution
confplot-0.1.1-py3-none-any.whl
(10.7 kB
view hashes)