跳到正文
原文
Lilian Weng:Lil'Log(RSS)·· 2017-07-22AI 评分35

用 RNN 预测股票价格(第二部分):引入股票代码嵌入

Predict Stock Prices Using RNN: Part 2

AI 导读

在 RNN 股票价格预测教程第二部分中,作者为第一部分搭建的循环神经网络加入股票代码嵌入向量,使其能同时处理多只股票。模型将股票代码映射为长度为 3 的嵌入向量,与每日价格输入拼接后送入 LSTM,替代 one-hot 编码以压缩表示并让相似股票共享模式。新增 embedding_size 和 stock_count 两个配置项,数据来自 Google Finance 历史股价。

来源:Lilian Weng:Lil'Log(RSS) · lilianweng.github.io