Schedule Sampling Seq2Seq -- Movie_Script Chatbot -- PyTorch

Objective

A chatbot which can learn different speaking types between male/female from movie scripts

Dataset

Cornell movie-dialogue corpus

• 220,579 conversational exchanges between 10,292 pairs of movie characters
• involves 9,035 characters from 617 movies

Model

1. Seq2Seq [1]

2. Scheduled Sampling for Seq2Seq [2]

Result

1. Loss

2. Chatting...

Reference paper

• [1] Sequence to Sequence Learning with Neural Networks, Ilya Sutskever, Oriol Vinyals, Quoc V. Le, 2014
• [2] Scheduled Sampling for Sequence Prediction with Recurrent Neural Networks, Samy Bengio, Oriol Vinyals, Navdeep Jaitly, Noam Shazeer, 2015