Some visualization utils mostly in Python
Visualization utilities for machine learning in python.
vis_confusion_matrix:
Visualize the confusion matrix in Python. Input: Confusion matrix and the class list in the same order as used in confusion matrix. To get the confusion matrix elements in the same order as in class_list use this list to generate the confusion matrix in scikit-learn package.
example usage:
from sklearn.metrics import confusion_matrix
cm = confusion_matrix(true_labels, pred_labels, class_list)
vis_confusion_matrix(cm, class_list)
You can easily change the color of the confusion matrix. I used 'pyplot.cm.bone'. If you want to look at other options see here: http://www.physics.ox.ac.uk/Users/msshin/science/code/matplotlib_cm/