我是 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)