---

作者

---

可下载资源

数据集介绍

我们使用的数据集涵盖了平安银行自2017年3月1日至2021年9月7日的股票历史行情数据。这些数据详细记录了每日的收盘价等重要信息，为我们提供了宝贵的分析材料。通过调用专业的数据接口——AKshare API，我们成功获取了这些高质量的数据，并将在后续的研究中充分利用这些数据来进行预测模型的构建和评估。

	import akshare as ak  

	import pandas as pd  

	  

 
	stock_zh_a_hist_df = ak.stock_zh_a_hist(symbol="000001", period="daily", start_date="20170301", end_date="20210907", adjust="")  

	  

	# 将数据保存到CSV文件  

	stock_zh_a_hist_df.to_csv("stock_data.csv", encoding="utf_8_sig")  # 推荐使用 utf_8_sig 以确保Excel可以正确打开  

	  

	# 如果你想查看数据集的一部分，可以使用head()方法  

	print(stock_zh_a_hist_df.head())

---

---

在训练阶段，我们采用编码器-解码器结构的神经网络模型进行联合训练。通过均方差函数（MSE）来计算模型输出序列与目标序列Y之间的误差，并使用Adam优化器进行迭代求解，以最小化误差。训练过程将进行500次迭代（epochs = 500），并在每个批次（batchsize = 32）的数据上进行参数更新。

使用其他模型进行对比

多种训练模型的训练均方差和验证均方差

模型训练结果

与RNN简单循环神经网络模型相比， LSTM具有能捕捉长期数据特征的优点，但是对于超长期的数据，仍旧无法进行很好的预测，而对于加入了Attention注意力机制的Transformer，正好可以解决这个难题，但是Transformer模型针对简单的数据集往往效果比不上LSTM，因为Transformer更容易导致过拟合，从而影响预测效果。

---

---

作者

• 理解为什么你需要能够预测股票价格的变动。
• 下载数据 – 使用从雅虎财经收集的股市数据
• 分割训练-测试数据，并进行数据归一化
• 应用单步预测技术。
• 讨论LSTM模型。
• 用当前的数据预测和可视化未来的股票市场

这就是时间序列模型的作用。你需要好的机器学习模型，它可以观察一连串数据的历史，并正确预测该序列的未来数据。

获取数据

你要利用航空公司的股票市场价格来进行预测，所以你把股票代码设置为 “AAL”。此外，你还定义了一个url\_string，它将返回一个JSON文件，其中包含航空公司过去20年的所有股市数据，以及一个file\_to_save，它是你保存数据的文件。

数据可视化

现在让我们来看看是什么样的数据。

---

---

plot(range(df.shape\[0\]),(df)/2.0)

现在你可以把训练数据和测试数据分开。训练数据将是时间序列的前4000个数据点，其余的将是测试数据。

train_data = mid\[:4000\]
test_data = mid\[4000:\]

标准化数据

现在你需要定义标准化来规范数据。 将训练和测试数据变化为[data\_size, num\_features]的维度。

将测试数据和训练数据相对于训练数据归一。

scaler = MinMaxScaler()

由于你先前的观察，即不同时间段的数据有不同的值范围，通过将全序列分割成窗口来标准化数据。如果你不这样做，早期的数据将接近于0，对学习过程不会有太大的价值。这里你选择了一个800的窗口大小。

# 用训练数据和平滑数据训练
window_size = 800
scaler.transform(train\_data\[di:di+window\_size,:\])

将数据重塑为[data_size]的形状。

# 重塑训练和测试数据
reshape(-1)

# 对测试数据进行标准化处理
scaler.transform(test_data).reshape(-1)

现在你可以使用指数移动平均线对数据进行平滑处理。

请注意，你应该只平滑训练数据。

# 现在进行指数移动平均平滑处理
# 所以数据会比原来的锯齿状数据有一个更平滑的曲线

  EMA = gamma\*train\[i\] + (1-gamma)\*EMA
  train\[i\] = EMA

首先，正常的平均数。

换句话说，你说t+1的预测是你在t到t-N的窗口内观察到的所有股票价格的平均值。

    pred.append(np.mean(train\[idx-window_size:idx\]))
    mse\_errors.append((std\_avg\[-1\]-train\[pred_idx\])**2)

MSE: 0.00418

看一下下面的平均结果。它与股票的实际行为相当接近。接下来，你将看到一个更准确的一步预测方法。

plt.plot(std\_avg\_pred)

plt.legend(fontsize=18)
plt.show()

那么，上面的图表（和MSE）说明了什么？

似乎对于非常短的预测（提前一天）来说，这个模型还不算太差。鉴于股票价格不会在一夜之间从0变化到100，这种行为是合理的。接下来，使用指数移动平均线。

指数移动平均线

你可能已经在互联网上看到一些文章，使用非常复杂的模型，并预测了几乎准确的股票市场行为。但是请注意! 这些只是视觉上的错觉，并不是由于学到了有用的东西。你将在下面看到如何用一个简单的平均法来复制这种行为。

在指数移动平均法中，你计算xt+1为。

其中  和  是在一段时间内保持的指数移动平均数值。.

上述公式基本上是计算t+1时间步长的指数移动平均线，并将其作为超前一步的预测。 γ决定最近的预测对EMA的贡献是什么。例如，γ=0.1只能得到当前值的10%进入EMA。因为你只取最近的一小部分，它允许保留你在平均数中很早看到的更早的值。请看下面用于预测向前一步的情况。

for idx in range(1,N):
    mean = mean\*dec + (1.0-de)\*train\[idx-1\]
    pred.append(mean)

MSE: 0.00003

plt.plot(mid_data)
plt.plot(pred)

如果指数式移动平均数这么好，为什么还需要更好的模型？

拟合结果很好，遵循真实的分布（并且由非常低的MSE证明）。实际上，仅凭第二天的股票市场价格是没有意义的。就我个人而言，我想要的不是第二天的确切股市价格，而是未来30天的股市价格是上涨还是下跌。尝试这样做，你会发现EMA方法的缺陷。

现在尝试在窗口中进行预测（比如你预测未来2天的窗口，而不是仅仅预测未来一天）。然后你会意识到EMA会有多大的误差。下面是一个例子。

预测超过一步的未来股价

我们假设数值，比如xt=0.4，EMA=0.5，γ=0.5

• 假设你得到的输出有以下公式

所以你有 

所以 

所以下一个预测Xt+2变成了

这就是 

或者在这个例子中, 

所以，无论你对未来进行多少步预测，你都会在所有未来的预测步中不断得到相同的答案。

你有一个解决方案，可以输出有用的信息，就是看看基于动量的算法。它们根据过去最近的数值是上升还是下降（不是准确的数值）进行预测。例如，它们会说第二天的价格可能会降低，如果过去几天的价格一直在下降，这听起来很合理。然而，我们使用一个更复杂的模型：LSTM模型。

这些模型已经在时间序列预测领域非常热门，因为它们在时间序列数据的建模方面非常出色。你将看到数据中是否真的隐藏有你可以利用的模式。

LSTM简介：对股票走势进行远期预测

长短期记忆模型是极其强大的时间序列模型。它们可以预测未来任意步。LSTM模块（或单元）有5个基本组成部分，使其能够对长期和短期数据进行建模。

• 单元状态（ct）–这代表了单元的内部记忆，它同时存储了短期记忆和长期记忆
• _隐藏状态_（ht）–这是根据当前输入、以前的_隐藏状态_和当前的单元输入计算出来的输出状态信息，你最终会用它来预测未来的股市价格。此外，隐藏状态可以决定只检索存储在单元状态中的短期或长期或两种类型的记忆来进行下一次预测。
• 输入门（it）–决定当前输入的信息有多少流向单元状态
• 遗忘门（ft）–决定有多少来自当前输入和前一个单元状态的信息流入当前单元状态
• 输出门（ot）–决定多少信息从当前单元状态流入隐藏状态，因此，如果需要，LSTM可以只挑选长期记忆或短期记忆和长期记忆。

TensorFlow为实现时间序列模型提供了一个不错的API（称为RNN API）。

数据生成器

你首先要实现一个数据生成器来训练你的模型。这个数据生成器将有一个名为.unroll\_batches(…)的方法，它将输出一组依次获得的num\_unrollings批次的输入数据，其中一个批次的数据大小为[batch_size, 1]。那么每一批输入数据都会有一个相应的输出数据批。

例如，如果num\_unrollings=3，batch\_size=4，一组unrolled批次。

• 输入数据
• 
• 输出数据:
• 

数据增强(_Data_ _Augmentation_)

数据增强(_Data_ _Augmentation_)又称为数据增广。另外，为了使你的模型稳健，你不会让x\_t的输出总是x\_t+1。相反，你将从x\_t+1,x\_t+2,…,xt+N的集合中随机抽取一个输出，其中N是一个小窗口大小。

这里你要做以下假设:

• x\_t+1,x\_t+2,…,xt+N不会彼此相距很远。

我个人认为这对股票走势预测来说是一个合理的假设。

下面你直观地说明一批数据是如何产生的。

定义超参数

在本节中，你将定义几个超参数。 D是输入的维度。这很简单，因为你把之前的股票价格作为输入，并预测下一个股票价格，这应该是1。

然后你有num\_unrollings，这是一个与用于优化LSTM模型的通过时间的反向传播（BPTT）有关的超参数。这表示你在一个优化步骤中考虑多少个连续的时间步骤。可以认为，不是通过查看单个时间步骤来优化模型，而是通过查看num\_unrollings时间步骤来优化网络。越大越好。

然后，你有batch_size。批量大小是指在一个时间步长中考虑多少个数据样本。

接下来你定义num_nodes，它代表每个单元中隐藏神经元的数量。你可以看到，在这个例子中，有三层LSTMs。

D = 1 # 数据的维度。因为你的数据是一维的，所以是1
unrollings = 50 # 未来的时间步数。
batch_size = 500 # 一个批次中的样本数
num_nodes = \[200,200,150\] # 我们所使用的深层LSTM堆栈中每一层的隐藏节点数量
n\_layers = len(num\_nodes) # 层的数量
dropout = 0.2 # 丢弃量

定义输入和输出

接下来你要为训练输入和标签定义占位符。你有一个输入占位符的列表，其中每个占位符都包含一个批次的数据。而列表中有num_unrollings占位符，这些占位符将被一次性用于一个优化步骤。

#输入数据
train\_inputs, train\_outputs = \[\],\[\]

定义LSTM和回归层的参数

你将有三层LSTM和一个线性回归层，用w和b表示，它采取最后一个长短期记忆单元的输出，并输出下一个时间步骤的预测。此外，你可以让dropout实现LSTM单元，因为它们可以提高性能，减少过拟合。

计算LSTM输出并将其传递到回归层以获得最终预测结果

在这一节中，你首先创建TensorFlow变量（c和h），这些变量将保持长短时记忆单元的状态和隐藏状态。然后，你将训练输入的列表转换为[unrollings, batch\_size, D]的形状。然后用ynamic\_rnn函数计算LSTM输出，并将输出分割成num张量列表。

# 创建单元格状态和隐藏状态变量保持LSTM的状态

for li in range(n):
  c.append(tf.Variable(tf.zeros(\[batch\_size, num\_nodes\[li\]\])))
  h.append(tf.Variable(tf.zeros(\[batch\_size, num\_nodes\[li\]\])))

# 做几次张量转换，因为函数dynamic_rnn要求输出必须是一种特定的格式。

损失计算和优化器

现在，要计算损失。对于每一批预测和真实输出，都要计算出平均平方误差。而你把所有这些均方差损失加在一起（不是平均）。最后，定义你要使用的优化器来优化神经网络。在这种情况下，你可以使用Adam，它是一个非常新的、表现良好的优化器。

# 在计算损失时，你需要注意准确的形式，因为你计算的是所有未滚动的步的损失
# 因此，取每个批的平均误差，并得到所有未滚动步的总和

range(n)\]):
  for ui in range(nums):
    loss += tf.mean(0.5*(splits\[ui\]-train\[ui\])**2)

预测相关的计算

在这里，你定义了预测相关的TensorFlow操作。首先，定义一个用于输入的占位符（sample\_inputs），然后与训练阶段类似，你定义预测的状态变量（sample\_c和sample\_h）。最后，你用dynamic\_rnn函数计算预测结果，然后通过回归层（w和b）发送输出。你还应该定义reset\_sample\_state操作，它可以重置单元状态和隐藏状态。

# 为预测阶段保持LSTM状态
for li in range(n_layers):
  sample\_c.append(tf.Variable(tf.zeros(\[1, num\_nodes\[li\]\]))
  sample\_h.append(tf.Variable(tf.zeros(\[1, num\_nodes\[li\]\])))

运行LSTM

在这里，你将训练并预测几个历时的股票价格走势，并观察预测结果是否随着时间的推移而变得更好或更差。

• 在时间序列上定义一个测试起点集（test\_points\_seq）来评估模型。
• 对于每个训练轮数
• 对于训练数据的完整序列长度
• 通过迭代测试点之前的num_unrollings数据点来更新LSTM状态

• 连续进行n\_predict\_once步骤的预测，将之前的预测作为当前输入。
• 计算预测的n\_predict\_once点与这些时间戳的真实股票价格之间的MSE损失
• 展开一组num_unrollings的批次
• 用未滚动的批次训练神经网络
• 计算平均训练损失
• 对于测试集的每个起点

…

预测可视化

你可以看到MSE损失是如何随着训练量的增加而下降的。这是一个好兆头，表明模型正在学习一些有用的东西。你可以将网络的MSE损失与你做标准平均时得到的MSE损失（0.004）进行比较。你可以看到，LSTM比标准平均法做得更好。而且你知道，标准平均法（虽然不完美）合理地遵循了真实的股票价格变动。

best_epoch = 28 # 用得到最佳结果的epoch


# 绘制预测值随时间变化的情况
# 绘制低α值的旧预测和高α值的新预测

        plt.plot(xval,yval)


# 预测你得到的最佳测试预测值
plt.plot(range(df.shap),mid_data)

    plt.plot(xval,yval)

虽然不完美，但LSTM似乎能够在大多数时候正确预测股票价格行为。请注意，你所做的预测大致在0和1.0的范围内（也就是说，不是真实的股票价格）。这没关系，因为你预测的是股票价格的走势，而不是价格本身。

总结

在本教程中，首先介绍了你为什么需要为股票价格建模的动机。接着是解释数据。然后两种平均技术，它们允许你对未来一步进行预测。接下来你看到，当你需要预测超过一步的未来时，这些方法是无用的。此后，讨论了如何使用LSTM来进行未来多步的预测。最后，将结果可视化，看到模型（虽然不完美）在正确预测股票价格走势方面相当出色。

在这里，我陈述一下本教程的几个收获。

2. 股票价格/走势预测是一项极其困难的任务。我个人认为，不应该把任何一个股票预测模型视为理所当然，盲目地依赖它们。然而模型可能在大多数时候能够正确预测股票价格走势，但不总是如此。
3. 不要被外面那些显示预测曲线与真实股票价格完全重合的文章所迷惑。这可以用一个简单的平均技术来复制，在实践中它是没有用的。更明智的做法是预测股票价格的变动。
4. 该模型的超参数对你获得的结果非常敏感。因此，要做的一件非常好的事情是对超参数运行一些超参数优化技术（例如，网格搜索/随机搜索）。下面我列出了一些最关键的超参数

• 优化器的学习率
• 层数和每层的隐藏单元的数量
• 优化器。我发现Adam的表现最好
• 模型的类型。你可以尝试GRU/标准LSTM和评估性能差异。