RNN学习预测飞机月流量,非常简单,不过自己练习的时候犯了两个错误,一是转Variable的时候放在了循环体内:
另一个是,因为用了gpu版,所有数据都要先.cuda()变成gpu版后才能使用,同时网络输出的结果也是gpu版,这时候直接把测试结果转成numpy用来绘图的时候报错了,需要先转成cpu版
我训练了5000次得到的效果:
分享快讯到朋友圈
领取专属 10元无门槛券
私享最新 技术干货