最简rnn_lstm模型python源码
1.源码
GitCode - 全球开发者的开源社区,开源代码托管平台
不到120行代码,参考了《深度学习与交通大数据实战》3.2节。注意这本书只能在京东等在线商城网购,才能拿到相应的数据集和源码。我的是在当地新华书店买的——买清华出版社,记得这个恶心的规定。
这是一个基于机器学习的时间预测。包含数据集生成,训练和预测。数据源直接模拟了一个10个周期的sin()函数,把前70%划分为训练集,建立了一个已知前2点,求下一点的训练集,并搭建相应的lstm模型。最终的效果(针对最终的30%数据,进行预测,左侧是真实的函数,右侧是预测):
稍稍对齐一下Y轴:
信号的相位信息基本一致,sin(x)特征基本保持,幅度有问题,然后时间序列预测最前面出现了一点异常。不知道有谁能消除掉它吗?