Schedule_Sampling_Seq2Seq_Movie_Script_Chatbot_PyTorch

KO | EN

gitlite — search

#javascript #python #hacktoberfest #react #ai #typescript #llm #go #golang #android #machine-learning #rust #deep-learning #linux

Schedule_Sampling_Seq2Seq_Movie_Script_Chatbot_PyTorch

★ 8 Open GitHub ↗

Build a chatbot based on gender with Seq2Seq and Schedule-Sampling Seq2Seq with PyTorch

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# 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](https://www.cs.cornell.edu/~cristian/Cornell_Movie-Dialogs_Corpus.html)
 - 220,579 conversational exchanges between 10,292 pairs of movie characters 
 - involves 9,035 characters from 617 movies
 
 ### Model
 1. Seq2Seq [[1]](https://papers.nips.cc/paper/5346-sequence-to-sequence-learning-with-neural-networks.pdf)
 <img src='http://4.bp.blogspot.com/-aArS0l1pjHQ/Vjj71pKAaEI/AAAAAAAAAxE/Nvy1FSbD_Vs/s1600/2TFstaticgraphic_alt-01.png' width="800" height="200">
 
 2. Scheduled Sampling for Seq2Seq [[2]](https://arxiv.org/abs/1506.03099)
 <img src='./pic/sche.png' width="400" height="300">
 
 ### Result
 1. Loss
 <img src='./pic/loss.png' width="400" height="200">
  
 2. Chatting...
 <img src='./pic/chat1.png' width="400" height="200">
 <img src='./pic/chat2.png' width="400" height="200">
 
 ## 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

Schedule_Sampling_Seq2Seq_Movie_Script_Chatbot_PyTorch

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README