python 训练象棋棋谱模型_使用Python训练回归模型并进行预测

使用Python训练回归模型并进行预测

回归分析是一种常见的统计方法，用于确定不同变量间的相互关系。在Excel中可以通过数据分析菜单中的回归功能快速完成。本篇文章将介绍在python中使用机器学习库sklearn建立简单回归模型的过程。

1. 准备工作

首先是开始前的准备工作，在创建回归模型的过程中我们需要使用以下几个库文件，他们分别为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

2. 读取并查看数据表

准备工作完成后，开始读取数据，这里我们使用了一组每日广告成本和点击量的数据。将这组数据读取到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()

3. 设置模型的自变量和因变量

在cost_and_click数据表中，我们将广告成本cost设置为自变量X，将广告点击量click设置为因变量Y。希望通过回归模型发现广告成本对于广告点击量的影响，以及两者间的关系。下面我们将X设置为cost，Y设置为click。并通过shape函数查看了两个变量的行