Make seq2seq for keras work.
The code includes:
- small dataset of movie scripts to train your models on
- preprocessor function to properly tokenize the data
- word2vec helpers to make use of gensim word2vec lib for extra flexibility
- train and predict function to harness the power of seq2seq
Warning
- The code has bugs, undoubtedly. Feel free to fix them and pull-request.
Papers
Nice picture
Setup&Run
git clone https://github.com/nicolas-ivanov/debug_seq2seq
cd debug_seq2seq
bash bin/setup.sh
python bin/train.py
or
python bin/test.py