facebookresearch / EmpatheticDialogues

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

data file #19

Closed yurui12138 closed 4 years ago

yurui12138 commented 4 years ago

Hello, I recently read your paper. Your work is very meaningful for the dialogue system, so I want to track your work. However, due to my limited technical level, I have some questions about the code and hope to get your advice. When using your code, how should the files in the data folder mentioned in the parameters be obtained? For example, how does "word dictionary" in "reddit dir" get?

EricMichaelSmith commented 4 years ago

Hi there! See my response at https://github.com/facebookresearch/EmpatheticDialogues/issues/16 - that should give you a sense of how to load in Reddit data in the proper format. If you want to use a BERT model, you may also want to take a look at the code in build_bert_dictionary() at https://github.com/facebookresearch/EmpatheticDialogues/blob/master/empchat/datasets/loader.py#L48 for how to create a BERT dictionary.

yurui12138 commented 4 years ago

Thank you very much for your reply!

EricMichaelSmith commented 4 years ago

Sure thing - happy to help!