[MRG] Modified the confusion matrix example