facebookresearch / ParlAI

A framework for training and evaluating AI models on a variety of openly available dialogue datasets.
https://parl.ai
MIT License
10.49k stars 2.1k forks source link

Multilight teacher #5043

Closed mojtaba-komeili closed 1 year ago

mojtaba-komeili commented 1 year ago

Patch description Adding a custom version of the multilight teachers to the ParlAI side. These teachers only focus on the conversation side of the dataset and do not have the graph elements like their LIGHT counterpart.

Testing steps Added teachers tests as well as running them manually:

parlai dd -t light_multiparty

Screenshot 2023-05-30 at 4 14 18 PM

parlai dd -t light_multiparty --add-location-to-context true --add-personas-to-context true --include-speaker-in-label false --add-speaker-to-context-end true --datatype valid

Screenshot 2023-05-30 at 4 14 47 PM

parlai dd -t light_multiparty:SpeakerPredictionTeacher

Screenshot 2023-05-30 at 4 13 30 PM