基于Scikit-learn的多元线性回归模型构建与验证

废话不多说,直接上代码

python 复制代码
# import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error
import joblib  # 用于保存和加载模型

# 1. 导入数据(假设你有一个包含特征和目标值的CSV文件)
# 例如,data.csv 中有若干特征列和一个名为 'target' 的目标列
data = pd.read_csv('FuelConsumptionCo21.csv')

# 2. 准备数据
# 分离特征和目标值
X = data.drop('target', axis=1)
y = data['target']

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

# 3. 创建并训练模型
model = LinearRegression()
model.fit(X_train, y_train)

print(model.coef_)
print(model.intercept_)

# 4. 进行预测并评估模型
y_pred = model.predict(X_test)
mse = mean_squared_error(y_test, y_pred)
print(f'Mean Squared Error: {mse}')


# 5. 保存模型
joblib.dump(model, 'linear_regression_model.pkl')

# 6. 加载模型(可选,验证模型是否正确保存和加载)
loaded_model = joblib.load('linear_regression_model.pkl')
loaded_y_pred = loaded_model.predict(X_test)
loaded_mse = mean_squared_error(y_test, loaded_y_pred)
print(f'Mean Squared Error after loading: {loaded_mse}')

原始数据是汽车 二氧化碳污染指标排放,数据如下

|------------|-----------|----------------------|---------------------|----------------------|--------------------------|--------|
| ENGINESIZE | CYLINDERS | FUELCONSUMPTION_CITY | FUELCONSUMPTION_HWY | FUELCONSUMPTION_COMB | FUELCONSUMPTION_COMB_MPG | target |
| 2 | 4 | 9.9 | 6.7 | 8.5 | 33 | 196 |
| 2.4 | 4 | 11.2 | 7.7 | 9.6 | 29 | 221 |
| 1.5 | 4 | 6 | 5.8 | 5.9 | 48 | 136 |
| 3.5 | 6 | 12.7 | 9.1 | 11.1 | 25 | 255 |
| 3.5 | 6 | 12.1 | 8.7 | 10.6 | 27 | 244 |
| 3.5 | 6 | 11.9 | 7.7 | 10 | 28 | 230 |
| 3.5 | 6 | 11.8 | 8.1 | 10.1 | 28 | 232 |
| 3.7 | 6 | 12.8 | 9 | 11.1 | 25 | 255 |
| 3.7 | 6 | 13.4 | 9.5 | 11.6 | 24 | 267 |
| 2.4 | 4 | 10.6 | 7.5 | 9.2 | 31 | 212 |
| 2.4 | 4 | 11.2 | 8.1 | 9.8 | 29 | 225 |
| 3.5 | 6 | 12.1 | 8.3 | 10.4 | 27 | 239 |
| 5.9 | 12 | 18 | 12.6 | 15.6 | 18 | 359 |
| 5.9 | 12 | 18 | 12.6 | 15.6 | 18 | 359 |
| 4.7 | 8 | 17.4 | 11.3 | 14.7 | 19 | 338 |
| 4.7 | 8 | 18.1 | 12.2 | 15.4 | 18 | 354 |
| 4.7 | 8 | 17.4 | 11.3 | 14.7 | 19 | 338 |
| 4.7 | 8 | 18.1 | 12.2 | 15.4 | 18 | 354 |
| 5.9 | 12 | 18 | 12.6 | 15.6 | 18 | 359 |
| 2 | 4 | 9.9 | 7.4 | 8.8 | 32 | 202 |
| 2 | 4 | 11.5 | 8.1 | 10 | 28 | 230 |
| 2 | 4 | 10.8 | 7.5 | 9.3 | 30 | 214 |

相关推荐
金融OG1 小时前
99.8 金融难点通俗解释:净资产收益率(ROE)
大数据·python·线性代数·机器学习·数学建模·金融·矩阵
Damon小智1 小时前
全面评测 DOCA 开发环境下的 DPU:性能表现、机器学习与金融高频交易下的计算能力分析
人工智能·机器学习·金融·边缘计算·nvidia·dpu·doca
赵孝正1 小时前
特征选择(机器学习)
人工智能·机器学习
Damon小智3 小时前
合合信息DocFlow产品解析与体验:人人可搭建的AI自动化单据处理工作流
图像处理·人工智能·深度学习·机器学习·ai·自动化·docflow
孤独且没人爱的纸鹤3 小时前
【机器学习】深入无监督学习分裂型层次聚类的原理、算法结构与数学基础全方位解读,深度揭示其如何在数据空间中构建层次化聚类结构
人工智能·python·深度学习·机器学习·支持向量机·ai·聚类
yuanbenshidiaos7 小时前
【大数据】机器学习----------强化学习机器学习阶段尾声
人工智能·机器学习
好评笔记12 小时前
AIGC视频生成模型:Stability AI的SVD(Stable Video Diffusion)模型
论文阅读·人工智能·深度学习·机器学习·计算机视觉·面试·aigc
坐吃山猪17 小时前
机器学习10-解读CNN代码Pytorch版
pytorch·机器学习·cnn
羊小猪~~18 小时前
深度学习基础--LSTM学习笔记(李沐《动手学习深度学习》)
人工智能·rnn·深度学习·学习·机器学习·gru·lstm