LSTM預測正弦波

這裏我想在MxNet中生成LSTM的教程用法，以Tensorflow爲例。在https://github.com/mouradmourafiq/tensorflow-lstm-regression/blob/master/lstm_sin.ipynb「（位置這是我的主要代碼LSTM預測正弦波

import mxnet as mx 
import numpy as np 
import pandas as pd 
import argparse 
import os 
import sys 
from data_processing import generate_data 
import logging 
head = '%(asctime)-15s %(message)s' 
logging.basicConfig(level=logging.DEBUG, format=head) 
TIMESTEPS = 3 
BATCH_SIZE = 100 
X, y = generate_data(np.sin, np.linspace(0, 100, 10000), TIMESTEPS, seperate=False) 
train_iter = mx.io.NDArrayIter(X['train'], y['train'], batch_size=BATCH_SIZE, shuffle=True, label_name='lro_label') 
eval_iter = mx.io.NDArrayIter(X['val'], y['val'], batch_size=BATCH_SIZE, shuffle=False) 
test_iter = mx.io.NDArrayIter(X['test'], batch_size=BATCH_SIZE, shuffle=False) 
num_layers = 3 
num_hidden = 50 

data = mx.sym.Variable('data') 
label = mx.sym.Variable('lro_label') 

stack = mx.rnn.SequentialRNNCell() 
for i in range(num_layers): 
    stack.add(mx.rnn.LSTMCell(num_hidden=num_hidden, prefix='lstm_l%d_'%i)) 
#stack.reset() 
outputs, states = stack.unroll(length=TIMESTEPS, 
           inputs=data, 
           layout='NTC', 
           merge_outputs=True) 

outputs = mx.sym.reshape(outputs, shape=(BATCH_SIZE, -1)) 
# purpose of fc1 was to make shape change to (batch_size, *), or label shape won't match LSTM unrolled output shape. 
outputs = mx.sym.FullyConnected(data=outputs, num_hidden=1, name='fc1') 
label = mx.sym.reshape(label, shape=(-1,)) 
outputs = mx.sym.LinearRegressionOutput(data=outputs, 
           label=label, 
           name='lro') 
contexts = mx.cpu(0) 
model = mx.mod.Module(symbol = outputs, 
        data_names = ['data'], 
        label_names = ['lro_label']) 
model.fit(train_iter, eval_iter, 
     optimizer_params = {'learning_rate':0.005}, 
     num_epoch=4, 
     batch_end_callback=mx.callback.Speedometer(BATCH_SIZE, 2))

此代碼運行，但train_accuracy是楠。的問題是如何使它正確？既然展開了形有sequence_length，它怎麼能比得上標記形狀？爲什麼我的FC1淨有意義嗎？

來源

2017-08-16 user2189731

的代碼在一般沒有問題，我終於做到了工作，運行時間較慢雖然比TF。不知道爲什麼。我可以在MXNet教程中舉一個例子作爲LSTM的開始例子，因爲我發現MXNet例子通常非常複雜。 – user2189731

通auto_reset=False到Speedometer回調，比如說，batch_end_callback=mx.callback.Speedometer(BATCH_SIZE, 2, auto_reset=False)，應該可以解決的NaN列車ACC。

來源

2018-01-10 18:34:09 Yizhi

LSTM預測正弦波

回答

相關問題