Skip to content

iwyoo/LSTM-autoencoder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 

Repository files navigation

LSTM-autoencoder

TensorFlow LSTM-autoencoder implementation

Usage

# hidden_num : the number of hidden units in each RNN-cell
# inputs : a list of tensor with size (batch_num x step_num x elem_num)
ae = LSTMAutoencoder(hidden_num, inputs)
...
sess.run(init)
...
sess.run(ae.train, feed_dict={input_variable:input_array,...})

Reference

Unsupervised Learning of Video Representations using LSTMs (Nitish Srivastava, Elman Mansimov, Ruslan Salakhutdinov)

http://arxiv.org/abs/1502.04681

About

TensorFlow LSTM-autoencoder implementation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages