[MRG] Modified the confusion matrix example

You need to create a function for that :-)
Since it's used only once.