Skip to content

Latest commit

 

History

History
13 lines (10 loc) · 367 Bytes

README.md

File metadata and controls

13 lines (10 loc) · 367 Bytes

Bidirectional GRU with attention mechanism on imdb sentimental analysis dataset.

This toy project has not been finished yet (i.e the evaluating function)

Prerequisite:

  1. Python 3.5+
  2. Tensorflow 1.0+

To run:

  1. Download dataset: https://s3.amazonaws.com/text-datasets/imdb.npz
  2. Set hypyerparameters in config.py
  3. run python imdb-bi-rnn-attention.py