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]...
