在火炬和数据结构中训练LSTM模型

时间:2015-11-01 05:45:28

标签: torch lstm

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

0 个答案:

没有答案