我正在使用10倍交叉验证,并根据准确性和精确度对模型进行评估。每个模型会生成10次混淆矩阵。谁能告诉我如何汇总混淆矩阵并计算准确性?
谢谢!
您可以如下使用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)
您需要提供有关数据格式的更多信息!
如果您使用的是张量流,则可以引用此source