Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Ready: Feature 139 confusion matrix #144

Commits on Mar 16, 2017

  1. Initial version of ConfusionMatrix visualizer using the matplotlib pc…

    …olormesh.
    
    - Allows for percent or raw count representation of the predictions
    - Implements heatmap with white=0, green=100%, and yellow-orange-red heatmap for everything else
    - Allows zooming in on confusion matrix using passed list of classes, with accurate %-of-all-true calculations
    - Tested for moderately large class numbers (30+)
    - Diagonal line indicates accurate predictions
    - Documentation added to docs/examples/methods.rst for one example matrix
    
    Suggested future improvements:
    - Resize font based on image size + class count
    - Allow custom color coding, including custom colors for _over and _under values (e.g. zero and 100%)
    - Vary text font color based on background color
    - While this branch currently adds an example to methods.rst, the examples/confusionMatrix.ipynb has additional examples using different of the passed parameters. This should probably also be exported as rst and added to the docs, but there was not an obvious place to put it so I am excluding that for now.
    
    Note this commit squashes all previous commits on this branch
    NealHumphrey committed Mar 16, 2017
    Configuration menu
    Copy the full SHA
    ceee7f8 View commit details
    Browse the repository at this point in the history

Commits on Mar 20, 2017

  1. Configuration menu
    Copy the full SHA
    23ac982 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    4b900ef View commit details
    Browse the repository at this point in the history

Commits on Mar 23, 2017

  1. Merge branch 'develop' into feature-139-confusion-matrix

    # Conflicts:
    #	examples/nealhumphrey/confusionMatrix.ipynb
    #	yellowbrick/classifier.py
    #	yellowbrick/utils.py
    NealHumphrey committed Mar 23, 2017
    Configuration menu
    Copy the full SHA
    4a84a3f View commit details
    Browse the repository at this point in the history
  2. Fixes one incorrect automatic merge issue and re-runs the two example…

    … notebooks to make sure they work.
    NealHumphrey committed Mar 23, 2017
    Configuration menu
    Copy the full SHA
    ce71fc6 View commit details
    Browse the repository at this point in the history