- 积分
- 5457
- 贡献
-
- 精华
- 在线时间
- 小时
- 注册时间
- 2018-1-24
- 最后登录
- 1970-1-1
|
发表于 2019-2-28 09:43:44
|
显示全部楼层
本帖最后由 黄裳 于 2019-2-28 10:02 编辑
你的第一个YTrain应该是{3 4 3 2 1}吧?另外LSTM的输入不是三维的数组(samples, times, features)吗?如果MiniBatchSize=1, 就是:
1. samples一个一个的输入
2. 然后计算这个输入后的cost function,
3. 然后更新权重,
接着输入samples里的第二个,重复上面3步的操作,直到你计算完所有samples, 如果设定epoch大于1,则再进行第二轮的更新权重,再重复3步操作,知道达到最后你设定的epochs的次数,其实总体看就是
for i in epochs:
1. samples按 size(MiniBatchSize) 大小整体输入(注:对于你设定的1来说,就是一个一个的样例输入)
2. 然后计算整个MiniBatchSize输入后的cost function,
3. 然后更新权重,
最后就得到你训练好的模型了。
另外,LSTM输入的是(samples, times, features),这个samples是都应该是都是一类吧, 怎么你的samples里的是不同站点呢?它怎么训练呢?就是说你想要用一个LSTM网络训练出多个站的温度演变,也不是不可以,但我觉得你这样放应该是错的。你这个里面每个站点都是一个时间序列,不同站点就是多个站的时间序列的数据,或者说其实是时空数据。
|
|