我是Torch的新手,我正在尝试实施LSTM。以下是我的代码。这是使用rnn库实现LSTM的正确方法还是我需要做的更多?
此外,输入LSTM时我的数据应该是什么样的?刚订购了2个输入和1个输出?
require 'rnn'
require 'nn'
-- Build Model
lm = nn.Sequential()
rnn = nn.Sequencer(nn.FastLSTM(2, 30))
lm:add(rnn)
lm:add(nn.Linear(30, 1))
-- Train Model on 'data'
criterion = nn.MSECriterion()
trainer = nn.StochasticGradient(lm, criterion)
trainer.learningRate = 0.01
trainer:train(data)