「数据游戏」：使用岭回归预测招商银行的股价

沈建明发布于2019-07-31 10:25 / 2189人阅读

摘要：作者，数据游戏优胜队伍成员前阵子报名参加了数据游戏比赛，题目是预测月号星期三招商银行的股价，截止时间是在月号星期天。

作者：Max，「数据游戏」优胜队伍成员

前阵子报名参加了「数据游戏」比赛，题目是预测5月15号（星期三）招商银行的股价，截止时间是在5月12号（星期天）。在本次预测中，我用到的是岭回归。

岭回归

岭回归是回归的一种，它解决回归中重大疑难问题：排除多重共线性，进行变量的选择，在存在共线性问题和病态数据偏多的研究中有较大的实用价值。按照度娘百科的解释：岭回归是一种专用于共线性数据分析的有偏估计回归方法，实质上是一种改良的最小二乘估计法，通过放弃最小二乘法的无偏性，以损失部分信息、降低精度为代价获得回归系数更为符合实际、更可靠的回归方法，对病态数据的拟合要强于最小二乘法。

多重共线性指线性回归模型中的解释变量之间由于存在精确相关关系或高度相关关系而使模型估计失真或难以估计准确。

由于岭回归涉及的数学知识比较复杂，且文章篇幅有限，这里不过多展开。

数据获取

本次数据是通过 Tushare 的 get_hist_data() 获取的。Tushare 是一个免费、开源的 python 财经数据接口包。python 安装 tushare 直接通过 pip install tushare 即可安装。

import tushare as ts
data = ts.get_hist_data("600848")

运行之后可以查看它的前后几行数据。按照 tushare 官方的说明，get_hist_data() 只能获取近3年的日线数据，而他的返回值的说明是这样的：

date：日期；

open：开盘价；

high：最高价；

close：收盘价；

low：最低价；

volume：成交量；

price_change：价格变动；

p_change：涨跌幅；

ma5：5日均价；

ma10：10日均价；

ma20:20日均价；

v_ma5:5日均量；

v_ma10:10日均量；

v_ma20:20日均量

均价的意思大概就是股票 n 天的成交价格或指数的平均值。均量则跟成交量有关。至于其他的返回值，应该是一下子就能明白的吧。在获得数据之后，我们查看一下描述性统计，通过data.describe()查看是否存在什么异常值或者缺失值。

这样看来似乎除了由于周末以及节假日不开盘导致的当天的数据缺失以外，并没有其他的缺失和异常。但是这里我们不考虑节假日的缺失值。

数据预处理

由于获取的数据是按日期降序排序，但本次预测跟时间序列有关，因此我们需要把顺序转一下，让它按照日期升序排序。

data1 = data[::-1]

处理完顺序之后，我们要做一下特征值的选择。由于 volume 以及均量的值很大，如果不进行处理的话，很可能对整体的预测造成不良影响。由于时间有限，而且考虑到运算的复杂度，这里我没有对这些特征进行处理，而是直接将它们去掉了。至于均价，我是按照自己的理解，和10日均价、20日均价相比，5日均价的范围没那么大，对近期的预测会比另外两个要好，因此保留5日均价。接着，我用 sklearn.model_selection 的 cross_val_score，分别查看除 "open", "close", "high", "low", "ma5" 以外的其他剩余属性对预测值的影响。发现、 "p_change"、"price_change"这两个属性对预测结果的影响不大，为了节省内存，增加运算速度，提高预测的准确性，也直接把它们去掉了。完了之后，查看前后三行数据。

data1 = data1[["open","high","low","ma5","close"]]
data1.head(3), data1.tail(3)

建模预测

由于提交截止日期是周日，预测的是周三，因此需要先对周一周二的信息进行预测。在这里我突然想到一个问题，是用前一天的所有数据来训练模型以预测当天的 close 比较准确，还是用当天除了 close 以外的其他数据来训练模型以训练当天的 close 比较准呢？为了验证这个问题，我分别对这两种方法做了实验。

为了减少代码量，定义了一个函数用以评估模型的错误率。

def get_score(X_train, y_train):
    ridge_score = np.sqrt(-cross_val_score(ridge, X_train, y_train, cv=10, scoring="neg_mean_squared_error"))
    return np.mean(ridge_score)

用前一天的所有数据来当训练集

y_train = data1["close"].values[1:]
X_train = data1.values[:-1]
score = get_score(X_train, y_train)

输出结果大约为0.469，这个错误率就比较大了，不太合理，更何况还要预测其他特征值作为测试数据。

用当天除了 close 以外的其他数据来当训练集

data2 = data1[:]
y_train = data2.pop("close").values
X_train = data2.values
score = get_score(X_train, y_train)

输出结果大约为0.183，跟第一个相比简直好多了。所以，就决定是你了！

接下来建模并把模型保存下来：

y_train = data1["close"]
X_train = data1[["open", "high", "low", "ma5"]]
close_model = ridge.fit(X_train, y_train)
joblib.dump(ridge, "close_model.m")

在预测之前呢，我们先拿训练集的后8组数据做一下测试，做个图看看：

scores = []
for x in X_train[-8:]:
    score = close_model.predict(np.array(x).reshape(1, -1))
    scores.append(score)
x = np.arange(8)
fig, axes = plt.subplots(1, 1, figsize=(13, 6))
axes.plot(scores)
axes.plot(y_train[-8:])
plt.xticks(x, data1.index[-8:].values, size=13, rotation=0)

看到这样子我还是相对比较放心的，不过，这个模型的训练值除了 close 以外的属性都是已知的，要预测三天后的还得预测前两天的测试值。

def get_model(s):
    y_train = data1[s].values[1:]
    X_train = data1.values[:-1]
    model = ridge.fit(X_train, y_train)
    return model
def get_results(X_test):
    attrs = ["open", "high", "low", "ma5"]
    results = []
    for attr in attrs:
        result = get_model(attr).predict(X_test)
        results.append(result)
    return results

接下来预测三天的股价：

X_test = data1[-1:].values
for i in range(3):
    results = get_results(X_test)
    close = close_model.predict(np.array(results).reshape(1, -1))
    print(close)
    results.append(close)
    X_test = np.array(results).reshape(1, -1)
print("5月15日招商银行关盘时的股价为：" + str(round(close[0], 2)))

输出结果为：

[33.46370029]
[33.4532047]
[33.43836721]

5月15日招商银行关盘时的股价为：33.44

总结

虽然预测结果是这样子，但感觉这样预测似乎很菜啊。毕竟预测的每个值都会有偏差，多个偏差累加起来就很多了，这让我有点害怕。不知道存不存在不预测其他值直接预测close的方法，或者说直接预测5月15号的而不用先预测13、14号的方法。虽然我知道有种算法是时间序列算法，但不是很懂。希望哪位大神看了能给我一些建议，指点迷津。

对于一个自学数据分析的在校学生，苦于没有项目经验，正好赶上这次的「数据游戏」，能利用此次机会操作一波真的很不错。既能学到东西，获胜的话还能得到奖品。

Ad Time

学习更多数据科学知识请关注微信公众号：read_csv

参与数据科学活动请加 QQ 群：759677734

服务器托管 idc机房托管数据挖掘的分类和预测高潜用户购买意向预测的数据使用云服务器的游戏数据分析回归分析

文章版权归作者所有，未经允许请勿转载,若此文章存在违规行为，您可以联系管理员删除。

转载请注明本文地址：https://www.ucloud.cn/yun/43782.html

「数据游戏」：使用岭回归预测招商银行的股价

摘要：作者，数据游戏优胜队伍成员前阵子报名参加了数据游戏比赛，题目是预测月号星期三招商银行的股价，截止时间是在月号星期天。作者：Max，「数据游戏」优胜队伍成员前阵子报名参加了「数据游戏」比赛，题目是预测5月15号（星期三）招商银行的股价，截止时间是在5月12号（星期天）。在本次预测中，我用到的是岭回归。岭回归岭回归是回归的一种，它解决回归中重大疑难问题：排除多重共线性，进行变量的选...

vboy1010 2019-06-26 18:59 评论0 收藏0
「数据游戏」：使用 ARIMA 算法预测三日后招商银行收盘价

摘要：作者天琼，数据游戏优胜队伍成员介绍本文整理记录了参与的一次小型数据分析竞赛数据游戏，竞赛目标是预测年月日股闭市时招商银行的股价。日发现的数据有错误，需要手工矫正日该数据恢复正常。而函数，是对样本外的数据进行预测。作者：天琼，「数据游戏」优胜队伍成员介绍本文整理记录了参与的一次小型数据分析竞赛「数据游戏」，竞赛目标是预测2019年5月15日A股闭市时招商银行600036的股价。主...

xcc3641 2019-07-31 10:25 评论0 收藏0
「数据游戏」：使用 ARIMA 算法预测三日后招商银行收盘价

摘要：作者天琼，数据游戏优胜队伍成员介绍本文整理记录了参与的一次小型数据分析竞赛数据游戏，竞赛目标是预测年月日股闭市时招商银行的股价。日发现的数据有错误，需要手工矫正日该数据恢复正常。而函数，是对样本外的数据进行预测。作者：天琼，「数据游戏」优胜队伍成员介绍本文整理记录了参与的一次小型数据分析竞赛「数据游戏」，竞赛目标是预测2019年5月15日A股闭市时招商银行600036的股价。主...

Nino 2019-06-26 18:59 评论0 收藏0
机器学习实战_线性回归&逻辑回归（二）

摘要：岭回归正则岭回归也称为正则化是线性回归的正则化版注意到这个正则项只有在训练过程中才会被加到代价函数。如果那此时的岭回归便变为了线性回归。线性模型的正则化正如我们在第一和第二章看到的那样，降低模型的过拟合的好方法是正则化这个模型（即限制它）：模型有越少的自由度，就越难以拟合数据。例如，正则化一个多项式模型，一个简单的方法就是减少多项式的阶数。对于一个线性模型，正则化的典型实现就是约...

yuanzhanghu 2019-06-26 16:00 评论0 收藏0
机器学习之多项式回归与模型泛化

摘要：还提供了，将多项式特征数据归一化和线性回归组合在了一起，大大方便的编程的过程。在机器学习算法中，主要的挑战来自方差，解决的方法主要有降低模型复杂度降维增加样本数使用验证集模型正则化。多项式回归多项式回归使用线性回归的基本思路非线性曲线如图： showImg(https://segmentfault.com/img/bVbkn4q?w=372&h=252); 假设曲线表达式为：$y...

huhud 2019-07-30 18:36 评论0 收藏0