从带有numpy或pandas的3x3混淆矩阵中提取常用度量的简单方法是通过计算准确率、精确率、召回率和F1值来评估分类模型的性能。
在numpy中,可以使用以下代码计算这些度量:
import numpy as np
confusion_matrix = np.array([[TP, FP, FN],
[FP, TN, FP],
[FN, FP, TN]])
accuracy = np.sum(np.diag(confusion_matrix)) / np.sum(confusion_matrix)
precision = confusion_matrix[0, 0] / np.sum(confusion_matrix[:, 0])
recall = confusion_matrix[0, 0] / np.sum(confusion_matrix[0, :])
f1_score = 2 * (precision * recall) / (precision + recall)
在pandas中,可以使用以下代码计算这些度量:
import pandas as pd
confusion_matrix = pd.DataFrame([[TP, FP, FN],
[FP, TN, FP],
[FN, FP, TN]])
accuracy = confusion_matrix.values.diagonal().sum() / confusion_matrix.values.sum()
precision = confusion_matrix.iloc[0, 0] / confusion_matrix.iloc[:, 0].sum()
recall = confusion_matrix.iloc[0, 0] / confusion_matrix.iloc[0, :].sum()
f1_score = 2 * (precision * recall) / (precision + recall)
腾讯云提供了多个与机器学习和数据分析相关的产品,例如腾讯云机器学习平台(https://cloud.tencent.com/product/tcmlp)、腾讯云数据分析(https://cloud.tencent.com/product/dla)等,可以帮助用户进行混淆矩阵的计算和模型性能评估。
领取专属 10元无门槛券
手把手带您无忧上云