Keras绘制混淆矩阵

最近在Keras里面用到了混淆矩阵，在这里记录一下。

下面这个代码可以利用Keras的model=Sequential()序列式模型产生的model直接用即可，比如下面这种序列模型（随便写一个）

model=Sequential()
model.add(Conv2D(filters=128,input_shape=(100,100,1),kernel_size=(6,2),strides=(2,1),activation='relu',))

model.add(BatchNormalization())     
model.add(Conv2D(filters=256,kernel_size=(6,1),strides=(3,1),padding='same',activation='relu'))

然后训练完成后的model直接用下面的代码块画一下就行了

from sklearn.metrics import confusion_matrix

import numpy as np
import matplotlib.pyplot as plt
import numpy as np
import itertools