Keras预测结果踩坑

57 阅读 0 评论 38 点赞

我是靠谱客的博主拉长花瓣，最近开发中收集的这篇文章主要介绍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

以上就是拉长花瓣为你收集整理的Keras预测结果踩坑的全部内容，希望文章能够帮你解决Keras预测结果踩坑所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。