A tensorflow implementation of sequenceto sequence (seq2seq) model from Tensorflow. Deployed on Telegram (@alphanovabot)
This is the full code for Nova conversational API, we implement Tensorflows Sequence to Sequence model to train a chatbot on the Cornell Movie Dialogue dataset. After training for a few hours, the bot is able to hold a fun conversation.
- numpy
- scipy
- six
- tensorflow (https://www.tensorflow.org/versions/r0.12/get_started/os_setup.html)
Use pip to install any missing dependencies
To train the bot, edit the seq2seq.ini
file so that mode is set to train like so
mode = train
then run the code like so
python main.py
To test the bot during or after training, edit the seq2seq.ini
file so that mode is set to test like so
mode = test
then run the code like so
python main.py
To deploy the bot after training, simply
python bot_main.py &
Contrbution regarding dataset checkpoints woukd be accepted as it is a great help for those without a GPU.
Simply create a pull request and mention the changes.
Copyright 2017 Aniruddh Chandratre
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.