如何在应用Stratified 10折交叉验证时获取python中所有混淆矩阵的聚合

Question

我正在使用10折交叉验证，并在准确性和精确度的基础上评估模型。为每个模型生成10次混淆矩阵。谁能告诉我如何聚合混淆矩阵并计算准确度？

谢谢！！

Answer 1

您可以按如下方式使用cross_val_predict函数，并将其结果用作confusion_matrix()参数。

from sklearn.metrics import confusion_matrix
from sklearn.model_selection import cross_val_predict

y_pred = cross_val_predict(clf, x, y, cv=5)
cm = confusion_matrix(y, y_pred)

Answer 2

您需要提供有关您拥有的数据格式的更多信息！

如果你正在使用tensorflow，你可以参考这个source