- This model is taken from torch chatbot tutorial. Run jupyter
Baseline/chatbot_tutorial.ipynb
Emotional Chatting Machine (ECM): Seq2Seq + Emotional Embedding + Internal memory + External memory (PyTorch Implementation)
- Open jupyter notebook
Baseline/jupyter/dialogue_preprocessing_ECM.ipynb
, get the dialogue data from google drive.
- Open directory
ecm/
, follow the readme, to run training on the datasetdaily dialogue
, change emotion category size to 7.