/TorchBot

Chatbot using PyTorch

Primary LanguageJupyter Notebook

Presenting TorchBot

Image

So this is basically a Chatbot based on Cornell Movie-Dialogs Corpus dataset built using PyTorch.Get the google colab here.

Steps involved

  • Handle loading and preprocessing of Cornell Movie-Dialogs Corpus dataset

  • Implement a sequence-to-sequence model with Luong attention mechanism(s)

  • Jointly train encoder and decoder models using mini-batches

  • Implement greedy-search decoding module

  • Interact with trained chatbot

References

https://github.com/ywk991112/pytorch-chatbot

https://github.com/spro/practical-pytorch/tree/master/seq2seq-translation

https://github.com/floydhub/textutil-preprocess-cornell-movie-corpus

https://pytorch.org/tutorials/beginner/chatbot_tutorial.html