Skip to content

Repository for dialogue models which enhance response diversity or coherence, coded in Pytorch.

Notifications You must be signed in to change notification settings

text-machine-lab/dialogue-pytorch

Repository files navigation

dialogue-pytorch

Repository for dialogue models which enhance response diversity or coherence, coded in Pytorch.

Contains shared code for multiple concurrent projects. First, this repository contains code for the processing of Reddit comments for use in training a dialogue model. Second, this repository contains code related to training a baseline sequence to sequence model (run_seq2seq.py). Finally, run_mismatch.py trains a classifier which diagnoses response coherence with respect to a dialogue history. With the --mismatch_path flag, the sequence-to-sequence can train using learned coherence features from the coherence classifier.

About

Repository for dialogue models which enhance response diversity or coherence, coded in Pytorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages