基于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 |

相关推荐
龙腾AI白云2 小时前
AI检索增强生成(RAG):解决大模型幻觉的核心落地技术
数据库·人工智能·机器学习·知识图谱
zy_destiny4 小时前
深度学习实战-基于YOLOv8的玉米雄穗目标检测:从数据集下载到训练部署全流程实录
yolo·目标检测·机器学习
禹凕5 小时前
机器学习之Selenium(Machina Learning about Selenium)
爬虫·python·selenium·测试工具·机器学习
一直在努力的小宁5 小时前
【阅读笔记】具身智能的真机数采,到了分水岭
人工智能·深度学习·机器学习·agent·具身智能·vlm·vln
YH行业报告分析6 小时前
2026 AI法律合同审查平台市场洞察:NLP与机器学习如何推动企业合规与合同流程智能化?
人工智能·机器学习·自然语言处理
田里的水稻6 小时前
FA_融合和滤波(FF)-误差状态卡尔曼滤波(ESKF)
人工智能·机器学习·机器人·自动驾驶
Leo.yuan8 小时前
企业可信Data Agent怎么建:可信分析智能体(Traceable Analytic Agent)的分析链路与验证机制
大数据·人工智能·机器学习
YOLO数据集集合9 小时前
树木检测数据集 | 树木检测 树种分类 航拍林业 森林监测9131期
人工智能·算法·机器学习·分类·数据挖掘·林业·树种分类
空奈qwq10 小时前
ANN 全连接神经网络入门指南:从神经元到深度学习的第一步
人工智能·python·深度学习·神经网络·算法·机器学习
CAE虚拟与现实10 小时前
PyTorch和scikit-learn的区别
人工智能·pytorch·scikit-learn