facebookresearch / EmpatheticDialogues

Dialogue model that produces empathetic responses when trained on the EmpatheticDialogues dataset.
Other
450 stars 63 forks source link

[WIP] Standalone code for EmpatheticDialogues #1

Closed EricMichaelSmith closed 5 years ago

EricMichaelSmith commented 5 years ago

This PR adds code to run the retrieval models of the EmpatheticDialogues paper in a standalone repo. Most of the original code for this was written by Hannah Rashkin: I've changed the code as needed, for instance to add in BERT support, support retrieval with different datasets as contexts, etc. I'll finish the README file and add sample commands once I test this stripped-down code more thoroughly.

EricMichaelSmith commented 5 years ago

Thanks @stephenroller for all of the incredibly useful comments! I've just finished making changes reflecting all of them