如何在KFold交叉验证的情况下显示平均分类报告和混淆矩阵



我正试图使用Keras ANN分类器在python中执行5倍交叉验证,我需要一个单一的平均分类报告和所有折叠的混淆矩阵。

我可以计算每个折叠的混淆矩阵和分类报告,并按照这里提供的代码片段显示它们。

问题:

如何计算所有折叠的平均混淆矩阵和平均报告?

混淆矩阵而言,它用于对每个折叠的混淆矩阵进行汇总,因为它们包含绝对值而不是百分比。然而,如果你想要一个平均的混淆矩阵,你应该创建一个相同形状的confusion_mat_avg变量,并只取所有折叠的每个单元的平均值。

关于平均报告,解决方案类似。下面的答案可能会给你一个更好的主意。

如何在python中计算具有10倍的K倍交叉验证的不平衡数据集的精度、召回率和f1分数。

最新更新