python多元线性回归预测混淆矩阵绘制
时间: 2023-07-02 10:04:00 浏览: 113
Python多元线性回归预测程序
5星 · 资源好评率100%
多元线性回归预测混淆矩阵绘制需要一些前置的知识和工具,以下是一个可以参考的代码示例:
```python
# 导入所需的库
import pandas as pd
import numpy as np
from sklearn.linear_model import LinearRegression
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
# 加载数据集
data = pd.read_csv('data.csv')
# 数据集拆分
X = data.iloc[:, :-1].values
y = data.iloc[:, -1].values
# 创建多元线性回归模型
model = LinearRegression().fit(X, y)
# 进行预测
y_pred = model.predict(X)
# 将预测结果转换成二分类的标签
y_pred = np.where(y_pred > 0.5, 1, 0)
# 计算混淆矩阵
cm = confusion_matrix(y, y_pred)
# 绘制混淆矩阵
plt.imshow(cm, cmap=plt.cm.Blues)
plt.title('Confusion matrix')
plt.colorbar()
plt.xticks([0, 1], ['0', '1'])
plt.yticks([0, 1], ['0', '1'])
plt.xlabel('Predicted label')
plt.ylabel('True label')
plt.show()
```
需要注意的是,以上代码仅是一个示例,具体的细节可能因为数据集的不同而有所不同。在实际使用中,需要根据实际情况对代码进行适当的修改。
阅读全文