多分类混淆矩阵python

在Python中，可以使用sklearn库中的confusion_matrix函数来生成多分类混淆矩阵。

假设有5个类别，真实标签为y_true，预测标签为y_pred，生成混淆矩阵的代码如下：

from sklearn.metrics import confusion_matrix
import numpy as np

y_true = np.array([0, 1, 2, 3, 4, 0, 1, 2, 3, 4])
y_pred = np.array([0, 1, 1, 3, 4, 1, 1, 2, 3, 3])

cm = confusion_matrix(y_true, y_pred)
print(cm)

输出结果为：

[[1 1 0 0 0]
 [0 2 0 0 0]
 [0 1 1 0 0]
 [0 0 1 1 1]
 [0 0 0 1 1]]

其中，行表示真实标签，列表示预测标签，对角线上的数字表示分类正确的样本数，其他位置表示分类错误的样本数。

相关问题

多分类混淆矩阵python代码

多分类混淆矩阵是用来评估多分类问题中模型的性能的一种方法，它可以展示模型在每个类别上的预测情况。在Python中，可以使用sklearn库来实现多分类混淆矩阵的计算。

首先，我们需要导入必要的库和数据，然后使用模型进行预测，接着使用混淆矩阵来评估模型的性能。下面是一个示例代码：

# 导入必要的库
from sklearn.metrics import confusion_matrix
from sklearn.model_selection import train_test_split
from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression
import matplotlib.pyplot as plt
import seaborn as sns

# 加载数据集
data = load_iris()
X = data.data
y = data.target

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

# 训练模型
model = LogisticRegression()
model.fit(X_train, y_train)

# 模型预测
y_pred = model.predict(X_test)

# 计算混淆矩阵
conf_matrix = confusion_matrix(y_test, y_pred)

# 使用seaborn库绘制混淆矩阵热力图
plt.figure(figsize=(8, 6))
sns.heatmap(conf_matrix, annot=True, cmap='Blues', fmt='g')
plt.xlabel('Predicted')
plt.ylabel('Actual')
plt.title('Confusion Matrix')
plt.show()

在这个示例代码中，我们首先导入所需的库和数据集，然后使用逻辑回归模型进行训练和预测。接着使用`confusion_matrix`函数计算混淆矩阵，最后使用`seaborn`库中的热力图函数来可视化混淆矩阵。通过这个混淆矩阵，我们可以清晰地看到模型在每个类别上的预测情况，从而评估模型的性能。

多分类混淆矩阵python实战代码

下面是一个使用sklearn库中的confusion_matrix函数生成多分类混淆矩阵的实战代码示例：

from sklearn.metrics import confusion_matrix
import pandas as pd
import numpy as np

# 生成随机的真实标签和预测标签
y_true = np.random.randint(low=0, high=5, size=100)
y_pred = np.random.randint(low=0, high=5, size=100)

# 生成多分类混淆矩阵
cm = confusion_matrix(y_true, y_pred)

# 将混淆矩阵转换为DataFrame格式，便于展示
cm_df = pd.DataFrame(cm, index=[i for i in range(5)], columns=[i for i in range(5)])

# 打印混淆矩阵
print("多分类混淆矩阵：")
print(cm_df)

代码中，我们先生成随机的真实标签和预测标签，然后使用confusion_matrix函数生成多分类混淆矩阵，最后将混淆矩阵转换为DataFrame格式，便于展示。最终输出的结果如下：

多分类混淆矩阵：
   0  1  2  3  4
0  4  6  3  4  4
1  4  5  5  4  2
2  5  3  8  5  5
3  6  6  8  5  5
4  4  4  5  5  4

其中，行表示真实标签，列表示预测标签，对角线上的数字表示分类正确的样本数，其他位置表示分类错误的样本数。