摘要:本篇文章将介绍在python中使用机器学习库sklearn建立简单回归模型的过程。
回归分析是一种常见的统计方法,用于确定不同变量间的相互关系。在Excel中可以通过数据分析菜单中的回归功能快速完成。本篇文章将介绍在python中使用机器学习库sklearn建立简单回归模型的过程。
准备工作
首先是开始前的准备工作,在创建回归模型的过程中我们需要使用以下几个库文件,他们分别为sklearn库,numpy库,pandas库和matplotlib库。sklearn库用于计算线性回归模型中的关键参数,并对模型进行检验。numpy和pandas库用于数据导入,创建数据表和一些基础的计算工作。matplotlib库用于绘制散点图。
#导入机器学习linear_model库 from sklearn import linear_model #导入交叉验证库 from sklearn import cross_validation #导入数值计算库 import numpy as np #导入科学计算库 import pandas as pd #导入图表库 import matplotlib.pyplot as plt 读取并查看数据表
准备工作完成后,开始读取数据,这里我们使用了一组每日广告成本和点击量的数据。将这组数据读取到python中并取名为cost_and_click。通过head函数查看数据表中前5行的内容。以下是数据读取和查看的代码和结果。
#读取数据并创建数据表,名称为cost_and_click cost_and_click=pd.DataFrame(pd.read_excel(cost_and_click.xlsx))
#查看数据表前5行的内容 cost_and_click.head()
设置模型的自变量和因变量