Skip to content

plot_confusion_matrix improvements #402

@JJLWHarrison

Description

@JJLWHarrison

In my view, plot_confusion_matrix() could be improved if:

  • There was the ability to show the coefficients of the normalised confusion matrix in addition to or instead of the absolute confusion matrix coefficients.
  • The full range of the provided colormap was used (rather than the use of alpha to make the text legible)
  • It was possible to show a colorbar
  • The default size depended on the number of classes

I've written an extension to this function which can accomplish these things and am following the quick contributor checklist to submit for comment/review.. :)

Examples:

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions