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
最后
以上就是拉长花瓣最近收集整理的关于Keras预测结果踩坑的全部内容，更多相关Keras预测结果踩坑内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复