【附代码数据】python机器学习回归实例代码.pdf
版权申诉
5星 · 超过95%的资源 107 浏览量
2022-05-06
12:22:11
上传
评论
收藏 389KB PDF 举报 
【原创】附代码数据
有问题到淘宝找“大数据部落”就可以了
python 机器学习回归实例代码
1. 问题介绍
数据来源： Advertising 数据集
https://www.kaggle.com/purbar/advertising-data/data
数据描述： 数据集包含了 200 个不同市场的产品销售额， 每个销售额对应 3 种广
告媒体，分别是 TV, radio 和 newspaper
任务描述： 分析广告媒体与销售额之间的关系， 基于广告媒体预算， 预测销售额
评价指标：销售额为连续值，为回归问题，可以采用均方误差作为评价指标
开发环境： Python3.6 + pandas + scikit-learn
2. 代码解释
（1） 导入相关模块
其中：pandas 用于数据处理， matplotlib.pyplot 用于画图， LinearRegression
是线性回归模型， train_test_split 用于划分数据集， mean_squared_error 用
于求均方误差
（2） 导入数据
导入 CSV表格中的 advertising 数据， head 查看前五行数据。
资源评论
zhuozhuowink2023-09-05资源很实用，对我启发很大，有很好的参考价值，内容详细。- m0_647781502023-04-16资源内容详细全面，与描述一致，对我很有用，有一定的使用价值。
