Keras预测结果踩坑

Keras里的pridect函数出来的预测结果为概率值，在画混淆矩阵，对结果可视化的时候不方便。针对预测结果的概率值，可以采用一个numpy的函数返回最大概率的索引，来将结果转化为类别。具体实例如下：

predictions = model.predict(test_generator )

y_test_pred = np.argmax(predictions, axis=1)

三分类的混淆矩阵结果如下：

[ 0  0  0]
 [ 0  0  0]
 [16  0 12]

Keras模型获取混淆矩阵参考链接：

从Keras模型获取混淆矩阵 - Thinbug