Skip to content

Bidirectional GRU with attention mechanism on imdb sentimental analysis dataset

Notifications You must be signed in to change notification settings

AllenCX/IMDB-RNN-Attention

Repository files navigation

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

About

Bidirectional GRU with attention mechanism on imdb sentimental analysis dataset

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published