tree: de31ad5c3393ac11ada9db5c35f5f0d6b6a66435 [path history] [tgz]
  1. README.md
  2. char_rnn.py
  3. imdb_data.py
  4. imdb_model.py
  5. imdb_train.py
examples/rnn/README.md

Train RNN model over IMDB dataset

Recurrent neural networks (RNN) are widely used for modelling sequential data, e.g., natural language sentences. This example describes how to implement a RNN application (or model) using SINGA's CUDNN RNN layers. We will use the LSTM model as an example to train on IMDB dataset.

Instructions

  • Prepare the dataset,

      python imdb_data.py
    
  • Start the training,

      python imdb_train.py