I've seen this confusion matrix example using your code, but I can't manage to find the specific color map that was used here. 