eisenjulian / neural-conversation-spanish

A neural conversation model implementation in Spanish built with Tensor Flow
MIT License
2 stars 4 forks source link
chatbot seq2seq-chatbot tensorflow

Neural Conversation Spanish

A neural conversation model implementation in Spanish built with Tensor Flow

Based on this paper this model uses movie subtitles in Spanish from www.opensubtitles.org as well a pre-trained word vectors from FastText to train a LSTM network in TensorFlow to achieve a conversation agent.

Set Up

Python installation with TensorFlow. Check out https://www.tensorflow.org/install/ Download the subtitle data from opensubtitles.org from here. It may take a while...

Train

python translate.py

Test

python translate.py --decode
> ¿Cómo estas?
> Bien