From 8fc3feca93b83271a41aa75933ad8608b35e80d8 Mon Sep 17 00:00:00 2001 From: Jimmy Wei Date: Fri, 10 Jun 2022 12:16:58 -0400 Subject: [PATCH] Friends dataset (#4568) * feat: add friends dataset for multiparty convo * feat: generate examples for all 6 main characters in the friends corpus by default * fix: remove unused file * fix: add testing * fix: add speaker label inside label * feat: add support for data folds; clean up code * fix: skip __MACOS folder for zipped files to avoid exception * fix: formatting with autoformat.sh * feat: add convenience teacher classes * feat: add command line option to specify list of characters * undo changes to build_data.py * cleanup * style fix --- parlai/tasks/friends/__init__.py | 5 + parlai/tasks/friends/agents.py | 193 ++++++++++++++++++ parlai/tasks/friends/build.py | 96 +++++++++ parlai/tasks/friends/test.py | 39 ++++ .../test/friends_all_characters_test.yml | 63 ++++++ .../test/friends_all_characters_train.yml | 60 ++++++ .../test/friends_all_characters_valid.yml | 73 +++++++ .../friends/test/friends_chandler_test.yml | 59 ++++++ .../friends/test/friends_chandler_train.yml | 58 ++++++ .../friends/test/friends_chandler_valid.yml | 72 +++++++ .../tasks/friends/test/friends_joey_test.yml | 59 ++++++ .../tasks/friends/test/friends_joey_train.yml | 58 ++++++ .../tasks/friends/test/friends_joey_valid.yml | 72 +++++++ .../friends/test/friends_monica_test.yml | 59 ++++++ .../friends/test/friends_monica_train.yml | 58 ++++++ .../friends/test/friends_monica_valid.yml | 71 +++++++ .../friends/test/friends_phoebe_test.yml | 59 ++++++ .../friends/test/friends_phoebe_train.yml | 59 ++++++ .../friends/test/friends_phoebe_valid.yml | 71 +++++++ .../friends/test/friends_rachel_test.yml | 59 ++++++ .../friends/test/friends_rachel_train.yml | 58 ++++++ .../friends/test/friends_rachel_valid.yml | 71 +++++++ .../tasks/friends/test/friends_ross_test.yml | 63 ++++++ .../tasks/friends/test/friends_ross_train.yml | 59 ++++++ .../tasks/friends/test/friends_ross_valid.yml | 71 +++++++ parlai/tasks/friends/test/friends_test.yml | 63 ++++++ parlai/tasks/friends/test/friends_train.yml | 60 ++++++ parlai/tasks/friends/test/friends_valid.yml | 73 +++++++ parlai/tasks/task_list.py | 11 + 29 files changed, 1872 insertions(+) create mode 100644 parlai/tasks/friends/__init__.py create mode 100644 parlai/tasks/friends/agents.py create mode 100644 parlai/tasks/friends/build.py create mode 100644 parlai/tasks/friends/test.py create mode 100644 parlai/tasks/friends/test/friends_all_characters_test.yml create mode 100644 parlai/tasks/friends/test/friends_all_characters_train.yml create mode 100644 parlai/tasks/friends/test/friends_all_characters_valid.yml create mode 100644 parlai/tasks/friends/test/friends_chandler_test.yml create mode 100644 parlai/tasks/friends/test/friends_chandler_train.yml create mode 100644 parlai/tasks/friends/test/friends_chandler_valid.yml create mode 100644 parlai/tasks/friends/test/friends_joey_test.yml create mode 100644 parlai/tasks/friends/test/friends_joey_train.yml create mode 100644 parlai/tasks/friends/test/friends_joey_valid.yml create mode 100644 parlai/tasks/friends/test/friends_monica_test.yml create mode 100644 parlai/tasks/friends/test/friends_monica_train.yml create mode 100644 parlai/tasks/friends/test/friends_monica_valid.yml create mode 100644 parlai/tasks/friends/test/friends_phoebe_test.yml create mode 100644 parlai/tasks/friends/test/friends_phoebe_train.yml create mode 100644 parlai/tasks/friends/test/friends_phoebe_valid.yml create mode 100644 parlai/tasks/friends/test/friends_rachel_test.yml create mode 100644 parlai/tasks/friends/test/friends_rachel_train.yml create mode 100644 parlai/tasks/friends/test/friends_rachel_valid.yml create mode 100644 parlai/tasks/friends/test/friends_ross_test.yml create mode 100644 parlai/tasks/friends/test/friends_ross_train.yml create mode 100644 parlai/tasks/friends/test/friends_ross_valid.yml create mode 100644 parlai/tasks/friends/test/friends_test.yml create mode 100644 parlai/tasks/friends/test/friends_train.yml create mode 100644 parlai/tasks/friends/test/friends_valid.yml diff --git a/parlai/tasks/friends/__init__.py b/parlai/tasks/friends/__init__.py new file mode 100644 index 00000000000..240697e3247 --- /dev/null +++ b/parlai/tasks/friends/__init__.py @@ -0,0 +1,5 @@ +#!/usr/bin/env python3 + +# Copyright (c) Facebook, Inc. and its affiliates. +# This source code is licensed under the MIT license found in the +# LICENSE file in the root directory of this source tree. diff --git a/parlai/tasks/friends/agents.py b/parlai/tasks/friends/agents.py new file mode 100644 index 00000000000..a321b0bb339 --- /dev/null +++ b/parlai/tasks/friends/agents.py @@ -0,0 +1,193 @@ +#!/usr/bin/env python3 + +# Copyright (c) Facebook, Inc. and its affiliates. +# This source code is licensed under the MIT license found in the +# LICENSE file in the root directory of this source tree. + +from typing import Optional +from parlai.core.opt import Opt +from parlai.core.teachers import DialogTeacher +from parlai.core.params import ParlaiParser +from .build import build +from collections import defaultdict +import jsonlines +from parlai.utils.data import DatatypeHelper + +import copy +import os + +START_TOKEN = '__START__' +SILENCE_TOKEN = '__SILENCE__' + + +def _path(opt, filename): + return os.path.join(opt['datapath'], 'Friends', filename) + + +class DefaultTeacher(DialogTeacher): + def __init__(self, opt, shared=None): + opt = copy.deepcopy(opt) + build(opt) + self.fold = DatatypeHelper.fold(opt['datatype']) + opt['datafile'] = _path(opt, self.fold + '.jsonl') + self.characters = opt['characters'].split(',') + self.character = opt['character'] + self.use_silence_token = opt['use_silence_token'] + self.silence_token = opt['silence_token'] + self.use_start_token = opt['use_start_token'] + self.start_token = opt['start_token'] + super().__init__(opt, shared) + + def setup_data(self, datafile): + conversations = defaultdict(list) + + with jsonlines.open(datafile) as reader: + for utterance in reader: + text = utterance['text'] + speaker = utterance['speaker'] + conversation_id = utterance['conversation_id'] + + conversations[conversation_id].append( + {"text": text, "speaker": speaker} + ) + + for conversation_id in conversations: + utterances = conversations[conversation_id] + characters = set( + [u['speaker'] for u in utterances if u['speaker'] in self.characters] + ) + characters_string = ','.join( + sorted(list(characters)) + ) # sorted to ensure same order across runs + last_utterance_index = len(utterances) - 1 + + for index, utterance in enumerate(utterances): + if index == 0: + if self.use_start_token: + context = self.start_token + + else: # skip the first utterance since there's no context + speaker = utterance['speaker'] + text = utterance['text'] + context = f'{speaker}: {text}' + continue + + speaker = utterance['speaker'] + text = utterance['text'] + + prev_context = context + context += '\n' + f'{speaker}: {text}' + + isConversationDone = index == last_utterance_index + + # By default, generate training examples for all 6 main characters. + # Otherwise only generate training examples for the chosen character. + if ( + self.character == 'All' and speaker in self.characters + ) or speaker == self.character: + yield { + "text": prev_context, + "label": f'{speaker}: {text}', + "characters": characters_string, + }, isConversationDone + elif self.use_silence_token: + yield { + "text": prev_context, + "label": f'{self.character}: {self.silence_token}', + "characters": characters_string, + }, isConversationDone + + @classmethod + def add_cmdline_args( + cls, parser: ParlaiParser, partial_opt: Optional[Opt] = None + ) -> ParlaiParser: + super().add_cmdline_args(parser, partial_opt) + agent = parser.add_argument_group('Friends Corpus Arguments') + agent.add_argument( + '--character', + type=str, + default='All', + choices=[ + 'All', + 'Rachel Green', + 'Monica Geller', + 'Phoebe Buffay', + 'Joey Tribbiani', + 'Chandler Bing', + 'Ross Geller', + ], + help='Which speaker labels to train on', + ) + agent.add_argument( + '--characters', + type=str, + default='Rachel Green,Monica Geller,Phoebe Buffay,Joey Tribbiani,Chandler Bing,Ross Geller', + help='A comma-separated list of characters to train on when `--character` == `All`', + ) + agent.add_argument( + '--use-silence-token', + type='bool', + default=True, + help='Use silence token to generate training example for sentences where the chosen speaker is not speaking. Defaults to True.', + ) + agent.add_argument( + '--silence-token', + type=str, + default=SILENCE_TOKEN, + help='The token to use to indicate the chosen speaker is silent. Defaults to __SILENCE__', + ) + agent.add_argument( + '--use-start-token', + type='bool', + default=False, + help='Use start token at the beginning of each conversation, and include the first sentence as a training example. Defaults to False.', + ) + agent.add_argument( + '--start-token', + type=str, + default=START_TOKEN, + help='The token to use to indicate the beginning of a conversation. Defaults to __START__', + ) + return parser + + +class AllCharactersTeacher(DefaultTeacher): + def __init__(self, opt, shared=None): + opt['character'] = 'All' + super().__init__(opt, shared) + + +class RachelTeacher(DefaultTeacher): + def __init__(self, opt, shared=None): + opt['character'] = 'Rachel Green' + super().__init__(opt, shared) + + +class MonicaTeacher(DefaultTeacher): + def __init__(self, opt, shared=None): + opt['character'] = 'Monica Geller' + super().__init__(opt, shared) + + +class PhoebeTeacher(DefaultTeacher): + def __init__(self, opt, shared=None): + opt['character'] = 'Phoebe Buffay' + super().__init__(opt, shared) + + +class JoeyTeacher(DefaultTeacher): + def __init__(self, opt, shared=None): + opt['character'] = 'Joey Tribbiani' + super().__init__(opt, shared) + + +class ChandlerTeacher(DefaultTeacher): + def __init__(self, opt, shared=None): + opt['character'] = 'Chandler Bing' + super().__init__(opt, shared) + + +class RossTeacher(DefaultTeacher): + def __init__(self, opt, shared=None): + opt['character'] = 'Ross Geller' + super().__init__(opt, shared) diff --git a/parlai/tasks/friends/build.py b/parlai/tasks/friends/build.py new file mode 100644 index 00000000000..76dbd1109ba --- /dev/null +++ b/parlai/tasks/friends/build.py @@ -0,0 +1,96 @@ +#!/usr/bin/env python3 + +# Copyright (c) Facebook, Inc. and its affiliates. +# This source code is licensed under the MIT license found in the +# LICENSE file in the root directory of this source tree. +# Download and build the data if it does not exist. + +import parlai.core.build_data as build_data +from parlai.core.build_data import DownloadableFile +import os +import jsonlines +from collections import defaultdict +from sklearn.model_selection import train_test_split + +RANDOM_SEED = 123 + +RESOURCES = [ + DownloadableFile( + 'http://zissou.infosci.cornell.edu/convokit/datasets/friends-corpus/friends-corpus.zip', + 'friends-corpus.zip', + '51ae80ce345212839d256b59b4982e9b40229ff6049115bd54d885a285d2b921', + zipped=True, + ) +] + + +def generate_folds(dpath): + """ + Generate Data Folds based on the scene id. + """ + datafile = os.path.join(dpath, 'friends-corpus/utterances.jsonl') + train_datafile = os.path.join(dpath, 'train.jsonl') + valid_datafile = os.path.join(dpath, 'valid.jsonl') + test_datafile = os.path.join(dpath, 'test.jsonl') + + # Load the dataset + conversations = defaultdict(list) + with jsonlines.open(datafile) as reader: + for utterance in reader: + text = utterance['text'] + speaker = utterance['speaker'] + conversation_id = utterance['conversation_id'] + + if speaker != 'TRANSCRIPT_NOTE': + conversations[conversation_id].append( + { + "text": text, + "speaker": speaker, + "conversation_id": conversation_id, + } + ) + + # Split the dataset into 80% train, 10% valid, 10% test + train, valid_and_test = train_test_split( + list(conversations.keys()), test_size=0.2, random_state=RANDOM_SEED + ) + valid, test = train_test_split( + valid_and_test, test_size=0.5, random_state=RANDOM_SEED + ) + + # Save the data folds into separate files + with jsonlines.open(train_datafile, mode='w') as writer: + for conversation_id in train: + for utterance in conversations[conversation_id]: + writer.write(utterance) + + with jsonlines.open(valid_datafile, mode='w') as writer: + for conversation_id in valid: + for utterance in conversations[conversation_id]: + writer.write(utterance) + + with jsonlines.open(test_datafile, mode='w') as writer: + for conversation_id in test: + for utterance in conversations[conversation_id]: + writer.write(utterance) + + +def build(opt): + dpath = os.path.join(opt['datapath'], 'Friends') + version = '1.00' + + if not build_data.built(dpath, version_string=version): + print('[building data: ' + dpath + ']') + if build_data.built(dpath): + # An older version exists, so remove these outdated files. + build_data.remove_dir(dpath) + build_data.make_dir(dpath) + + # Download the data. + for downloadable_file in RESOURCES: + downloadable_file.download_file(dpath) + + generate_folds(dpath) + + # Mark the data as built. + build_data.mark_done(dpath, version_string=version) diff --git a/parlai/tasks/friends/test.py b/parlai/tasks/friends/test.py new file mode 100644 index 00000000000..ca23ab6589b --- /dev/null +++ b/parlai/tasks/friends/test.py @@ -0,0 +1,39 @@ +#!/usr/bin/env python3 + +# Copyright (c) Facebook, Inc. and its affiliates. +# This source code is licensed under the MIT license found in the +# LICENSE file in the root directory of this source tree. + +from parlai.utils.testing import AutoTeacherTest # noqa: F401 + + +class TestDefaultTeacher(AutoTeacherTest): + task = 'friends' + + +class TestAllCharactersTeacher(AutoTeacherTest): + task = 'friends:all_characters' + + +class TestRachelTeacher(AutoTeacherTest): + task = 'friends:rachel' + + +class TestMonicaTeacher(AutoTeacherTest): + task = 'friends:monica' + + +class TestPhoebeTeacher(AutoTeacherTest): + task = 'friends:phoebe' + + +class TestJoeyTeacher(AutoTeacherTest): + task = 'friends:joey' + + +class TestChandlerTeacher(AutoTeacherTest): + task = 'friends:chandler' + + +class TestRossTeacher(AutoTeacherTest): + task = 'friends:ross' diff --git a/parlai/tasks/friends/test/friends_all_characters_test.yml b/parlai/tasks/friends/test/friends_all_characters_test.yml new file mode 100644 index 00000000000..d39929566f2 --- /dev/null +++ b/parlai/tasks/friends/test/friends_all_characters_test.yml @@ -0,0 +1,63 @@ +acts: +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Rachel Green: Ooy.' + id: friends:all_characters + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what + is life without love?' + id: friends:all_characters + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Rachel Green: Oh my god, are we supposed to answer?' + id: friends:all_characters + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of + events, which I''m still fine with by the way. E is for how extremely normal + I find it. That you two are together. And now one day you might get married + and have children of your own.' + id: friends:all_characters + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: Dude, are you okay?' + id: friends:all_characters + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer? + + Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of events, + which I''m still fine with by the way. E is for how extremely normal I find + it. That you two are together. And now one day you might get married and have + children of your own.' +num_episodes: 302 +num_examples: 5774 diff --git a/parlai/tasks/friends/test/friends_all_characters_train.yml b/parlai/tasks/friends/test/friends_all_characters_train.yml new file mode 100644 index 00000000000..57d216b8072 --- /dev/null +++ b/parlai/tasks/friends/test/friends_all_characters_train.yml @@ -0,0 +1,60 @@ +acts: +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:all_characters + labels: + - 'Ross Geller: Hi!' + text: 'Phoebe Buffay: Oh hey! Wait up!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:all_characters + labels: + - 'Phoebe Buffay: Congratulations! I didn''t want to say anything in front of + Joey ''cause I didn''t know if he knew yet.' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:all_characters + labels: + - 'Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:all_characters + labels: + - 'Phoebe Buffay: No! No, that you and Rachel are engaged!' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:all_characters + labels: + - 'Ross Geller: What?' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier. + + Phoebe Buffay: No! No, that you and Rachel are engaged!' +num_episodes: 2427 +num_examples: 46610 diff --git a/parlai/tasks/friends/test/friends_all_characters_valid.yml b/parlai/tasks/friends/test/friends_all_characters_valid.yml new file mode 100644 index 00000000000..4e565bc3259 --- /dev/null +++ b/parlai/tasks/friends/test/friends_all_characters_valid.yml @@ -0,0 +1,73 @@ +acts: +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: Chandler, if it really hurts that bad you should just tell + her.' + id: friends:all_characters + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was!' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' + id: friends:all_characters + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Ross Geller: Hey.' + id: friends:all_characters + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' + id: friends:all_characters + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Ross Geller: Sorry.' + id: friends:all_characters + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey. + + Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' +num_episodes: 308 +num_examples: 5855 diff --git a/parlai/tasks/friends/test/friends_chandler_test.yml b/parlai/tasks/friends/test/friends_chandler_test.yml new file mode 100644 index 00000000000..62b63c545a1 --- /dev/null +++ b/parlai/tasks/friends/test/friends_chandler_test.yml @@ -0,0 +1,59 @@ +acts: +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Chandler Bing: __SILENCE__' + id: friends:chandler + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Chandler Bing: __SILENCE__' + id: friends:chandler + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Chandler Bing: __SILENCE__' + id: friends:chandler + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Chandler Bing: __SILENCE__' + id: friends:chandler + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Chandler Bing: __SILENCE__' + id: friends:chandler + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer? + + Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of events, + which I''m still fine with by the way. E is for how extremely normal I find + it. That you two are together. And now one day you might get married and have + children of your own.' +num_episodes: 302 +num_examples: 5774 diff --git a/parlai/tasks/friends/test/friends_chandler_train.yml b/parlai/tasks/friends/test/friends_chandler_train.yml new file mode 100644 index 00000000000..dd9a0cc5f81 --- /dev/null +++ b/parlai/tasks/friends/test/friends_chandler_train.yml @@ -0,0 +1,58 @@ +acts: +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:chandler + labels: + - 'Chandler Bing: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:chandler + labels: + - 'Chandler Bing: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:chandler + labels: + - 'Chandler Bing: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:chandler + labels: + - 'Chandler Bing: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:chandler + labels: + - 'Chandler Bing: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier. + + Phoebe Buffay: No! No, that you and Rachel are engaged!' +num_episodes: 2427 +num_examples: 46610 diff --git a/parlai/tasks/friends/test/friends_chandler_valid.yml b/parlai/tasks/friends/test/friends_chandler_valid.yml new file mode 100644 index 00000000000..669d9f4212d --- /dev/null +++ b/parlai/tasks/friends/test/friends_chandler_valid.yml @@ -0,0 +1,72 @@ +acts: +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Chandler Bing: __SILENCE__' + id: friends:chandler + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was!' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' + id: friends:chandler + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Chandler Bing: __SILENCE__' + id: friends:chandler + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Chandler Bing: __SILENCE__' + id: friends:chandler + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Chandler Bing: __SILENCE__' + id: friends:chandler + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey. + + Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' +num_episodes: 308 +num_examples: 5855 diff --git a/parlai/tasks/friends/test/friends_joey_test.yml b/parlai/tasks/friends/test/friends_joey_test.yml new file mode 100644 index 00000000000..ddd300ca695 --- /dev/null +++ b/parlai/tasks/friends/test/friends_joey_test.yml @@ -0,0 +1,59 @@ +acts: +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: __SILENCE__' + id: friends:joey + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: __SILENCE__' + id: friends:joey + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: __SILENCE__' + id: friends:joey + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: __SILENCE__' + id: friends:joey + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: Dude, are you okay?' + id: friends:joey + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer? + + Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of events, + which I''m still fine with by the way. E is for how extremely normal I find + it. That you two are together. And now one day you might get married and have + children of your own.' +num_episodes: 302 +num_examples: 5774 diff --git a/parlai/tasks/friends/test/friends_joey_train.yml b/parlai/tasks/friends/test/friends_joey_train.yml new file mode 100644 index 00000000000..520971ba273 --- /dev/null +++ b/parlai/tasks/friends/test/friends_joey_train.yml @@ -0,0 +1,58 @@ +acts: +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:joey + labels: + - 'Joey Tribbiani: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:joey + labels: + - 'Joey Tribbiani: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:joey + labels: + - 'Joey Tribbiani: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:joey + labels: + - 'Joey Tribbiani: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:joey + labels: + - 'Joey Tribbiani: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier. + + Phoebe Buffay: No! No, that you and Rachel are engaged!' +num_episodes: 2427 +num_examples: 46610 diff --git a/parlai/tasks/friends/test/friends_joey_valid.yml b/parlai/tasks/friends/test/friends_joey_valid.yml new file mode 100644 index 00000000000..463b97dda08 --- /dev/null +++ b/parlai/tasks/friends/test/friends_joey_valid.yml @@ -0,0 +1,72 @@ +acts: +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: Chandler, if it really hurts that bad you should just tell + her.' + id: friends:joey + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was!' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: __SILENCE__' + id: friends:joey + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: __SILENCE__' + id: friends:joey + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' + id: friends:joey + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: __SILENCE__' + id: friends:joey + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey. + + Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' +num_episodes: 308 +num_examples: 5855 diff --git a/parlai/tasks/friends/test/friends_monica_test.yml b/parlai/tasks/friends/test/friends_monica_test.yml new file mode 100644 index 00000000000..b9d3bd5c427 --- /dev/null +++ b/parlai/tasks/friends/test/friends_monica_test.yml @@ -0,0 +1,59 @@ +acts: +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer? + + Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of events, + which I''m still fine with by the way. E is for how extremely normal I find + it. That you two are together. And now one day you might get married and have + children of your own.' +num_episodes: 302 +num_examples: 5774 diff --git a/parlai/tasks/friends/test/friends_monica_train.yml b/parlai/tasks/friends/test/friends_monica_train.yml new file mode 100644 index 00000000000..9169177d8a4 --- /dev/null +++ b/parlai/tasks/friends/test/friends_monica_train.yml @@ -0,0 +1,58 @@ +acts: +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:monica + labels: + - 'Monica Geller: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:monica + labels: + - 'Monica Geller: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:monica + labels: + - 'Monica Geller: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:monica + labels: + - 'Monica Geller: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:monica + labels: + - 'Monica Geller: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier. + + Phoebe Buffay: No! No, that you and Rachel are engaged!' +num_episodes: 2427 +num_examples: 46610 diff --git a/parlai/tasks/friends/test/friends_monica_valid.yml b/parlai/tasks/friends/test/friends_monica_valid.yml new file mode 100644 index 00000000000..09372975698 --- /dev/null +++ b/parlai/tasks/friends/test/friends_monica_valid.yml @@ -0,0 +1,71 @@ +acts: +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was!' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Monica Geller: __SILENCE__' + id: friends:monica + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey. + + Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' +num_episodes: 308 +num_examples: 5855 diff --git a/parlai/tasks/friends/test/friends_phoebe_test.yml b/parlai/tasks/friends/test/friends_phoebe_test.yml new file mode 100644 index 00000000000..a12207f0ba9 --- /dev/null +++ b/parlai/tasks/friends/test/friends_phoebe_test.yml @@ -0,0 +1,59 @@ +acts: +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer? + + Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of events, + which I''m still fine with by the way. E is for how extremely normal I find + it. That you two are together. And now one day you might get married and have + children of your own.' +num_episodes: 302 +num_examples: 5774 diff --git a/parlai/tasks/friends/test/friends_phoebe_train.yml b/parlai/tasks/friends/test/friends_phoebe_train.yml new file mode 100644 index 00000000000..0b7fd89fa4a --- /dev/null +++ b/parlai/tasks/friends/test/friends_phoebe_train.yml @@ -0,0 +1,59 @@ +acts: +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:phoebe + labels: + - 'Phoebe Buffay: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:phoebe + labels: + - 'Phoebe Buffay: Congratulations! I didn''t want to say anything in front of + Joey ''cause I didn''t know if he knew yet.' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:phoebe + labels: + - 'Phoebe Buffay: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:phoebe + labels: + - 'Phoebe Buffay: No! No, that you and Rachel are engaged!' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:phoebe + labels: + - 'Phoebe Buffay: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier. + + Phoebe Buffay: No! No, that you and Rachel are engaged!' +num_episodes: 2427 +num_examples: 46610 diff --git a/parlai/tasks/friends/test/friends_phoebe_valid.yml b/parlai/tasks/friends/test/friends_phoebe_valid.yml new file mode 100644 index 00000000000..60be934c322 --- /dev/null +++ b/parlai/tasks/friends/test/friends_phoebe_valid.yml @@ -0,0 +1,71 @@ +acts: +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was!' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Phoebe Buffay: __SILENCE__' + id: friends:phoebe + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey. + + Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' +num_episodes: 308 +num_examples: 5855 diff --git a/parlai/tasks/friends/test/friends_rachel_test.yml b/parlai/tasks/friends/test/friends_rachel_test.yml new file mode 100644 index 00000000000..0f598bc3714 --- /dev/null +++ b/parlai/tasks/friends/test/friends_rachel_test.yml @@ -0,0 +1,59 @@ +acts: +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Rachel Green: Ooy.' + id: friends:rachel + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Rachel Green: __SILENCE__' + id: friends:rachel + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Rachel Green: Oh my god, are we supposed to answer?' + id: friends:rachel + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Rachel Green: __SILENCE__' + id: friends:rachel + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Rachel Green: __SILENCE__' + id: friends:rachel + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer? + + Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of events, + which I''m still fine with by the way. E is for how extremely normal I find + it. That you two are together. And now one day you might get married and have + children of your own.' +num_episodes: 302 +num_examples: 5774 diff --git a/parlai/tasks/friends/test/friends_rachel_train.yml b/parlai/tasks/friends/test/friends_rachel_train.yml new file mode 100644 index 00000000000..c6f63adcdc0 --- /dev/null +++ b/parlai/tasks/friends/test/friends_rachel_train.yml @@ -0,0 +1,58 @@ +acts: +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:rachel + labels: + - 'Rachel Green: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:rachel + labels: + - 'Rachel Green: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:rachel + labels: + - 'Rachel Green: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:rachel + labels: + - 'Rachel Green: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:rachel + labels: + - 'Rachel Green: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier. + + Phoebe Buffay: No! No, that you and Rachel are engaged!' +num_episodes: 2427 +num_examples: 46610 diff --git a/parlai/tasks/friends/test/friends_rachel_valid.yml b/parlai/tasks/friends/test/friends_rachel_valid.yml new file mode 100644 index 00000000000..72f58d81928 --- /dev/null +++ b/parlai/tasks/friends/test/friends_rachel_valid.yml @@ -0,0 +1,71 @@ +acts: +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Rachel Green: __SILENCE__' + id: friends:rachel + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was!' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Rachel Green: __SILENCE__' + id: friends:rachel + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Rachel Green: __SILENCE__' + id: friends:rachel + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Rachel Green: __SILENCE__' + id: friends:rachel + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Rachel Green: __SILENCE__' + id: friends:rachel + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey. + + Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' +num_episodes: 308 +num_examples: 5855 diff --git a/parlai/tasks/friends/test/friends_ross_test.yml b/parlai/tasks/friends/test/friends_ross_test.yml new file mode 100644 index 00000000000..ef9f12c4772 --- /dev/null +++ b/parlai/tasks/friends/test/friends_ross_test.yml @@ -0,0 +1,63 @@ +acts: +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Ross Geller: __SILENCE__' + id: friends:ross + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what + is life without love?' + id: friends:ross + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Ross Geller: __SILENCE__' + id: friends:ross + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of + events, which I''m still fine with by the way. E is for how extremely normal + I find it. That you two are together. And now one day you might get married + and have children of your own.' + id: friends:ross + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Ross Geller: __SILENCE__' + id: friends:ross + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer? + + Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of events, + which I''m still fine with by the way. E is for how extremely normal I find + it. That you two are together. And now one day you might get married and have + children of your own.' +num_episodes: 302 +num_examples: 5774 diff --git a/parlai/tasks/friends/test/friends_ross_train.yml b/parlai/tasks/friends/test/friends_ross_train.yml new file mode 100644 index 00000000000..2ea6f69c3a5 --- /dev/null +++ b/parlai/tasks/friends/test/friends_ross_train.yml @@ -0,0 +1,59 @@ +acts: +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:ross + labels: + - 'Ross Geller: Hi!' + text: 'Phoebe Buffay: Oh hey! Wait up!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:ross + labels: + - 'Ross Geller: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:ross + labels: + - 'Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:ross + labels: + - 'Ross Geller: __SILENCE__' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends:ross + labels: + - 'Ross Geller: What?' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier. + + Phoebe Buffay: No! No, that you and Rachel are engaged!' +num_episodes: 2427 +num_examples: 46610 diff --git a/parlai/tasks/friends/test/friends_ross_valid.yml b/parlai/tasks/friends/test/friends_ross_valid.yml new file mode 100644 index 00000000000..19940659cff --- /dev/null +++ b/parlai/tasks/friends/test/friends_ross_valid.yml @@ -0,0 +1,71 @@ +acts: +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Ross Geller: __SILENCE__' + id: friends:ross + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was!' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Ross Geller: __SILENCE__' + id: friends:ross + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Ross Geller: Hey.' + id: friends:ross + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Ross Geller: __SILENCE__' + id: friends:ross + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Ross Geller: Sorry.' + id: friends:ross + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey. + + Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' +num_episodes: 308 +num_examples: 5855 diff --git a/parlai/tasks/friends/test/friends_test.yml b/parlai/tasks/friends/test/friends_test.yml new file mode 100644 index 00000000000..ddde2aedacc --- /dev/null +++ b/parlai/tasks/friends/test/friends_test.yml @@ -0,0 +1,63 @@ +acts: +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Rachel Green: Ooy.' + id: friends + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what + is life without love?' + id: friends + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy.' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Rachel Green: Oh my god, are we supposed to answer?' + id: friends + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of + events, which I''m still fine with by the way. E is for how extremely normal + I find it. That you two are together. And now one day you might get married + and have children of your own.' + id: friends + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer?' +- - characters: Joey Tribbiani,Rachel Green,Ross Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: Dude, are you okay?' + id: friends + text: 'Ross Geller: Everyone? I would like to make a toast to Rachel and Joey. + + Rachel Green: Ooy. + + Ross Geller: And to love. Ah, love. L-O-V-E, love. L is for life. And what is + life without love? + + Rachel Green: Oh my god, are we supposed to answer? + + Ross Geller: O is for "oh, wow!" The V is for this very surprising turn of events, + which I''m still fine with by the way. E is for how extremely normal I find + it. That you two are together. And now one day you might get married and have + children of your own.' +num_episodes: 302 +num_examples: 5774 diff --git a/parlai/tasks/friends/test/friends_train.yml b/parlai/tasks/friends/test/friends_train.yml new file mode 100644 index 00000000000..045b2bc10c9 --- /dev/null +++ b/parlai/tasks/friends/test/friends_train.yml @@ -0,0 +1,60 @@ +acts: +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends + labels: + - 'Ross Geller: Hi!' + text: 'Phoebe Buffay: Oh hey! Wait up!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends + labels: + - 'Phoebe Buffay: Congratulations! I didn''t want to say anything in front of + Joey ''cause I didn''t know if he knew yet.' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi!' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends + labels: + - 'Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends + labels: + - 'Phoebe Buffay: No! No, that you and Rachel are engaged!' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier.' +- - characters: Phoebe Buffay,Ross Geller + episode_done: false + id: friends + labels: + - 'Ross Geller: What?' + text: 'Phoebe Buffay: Oh hey! Wait up! + + Ross Geller: Hi! + + Phoebe Buffay: Congratulations! I didn''t want to say anything in front of Joey + ''cause I didn''t know if he knew yet. + + Ross Geller: What, that we had a baby? Come on let''s give him a little credit, + although, he did eat a piece of plastic fruit earlier. + + Phoebe Buffay: No! No, that you and Rachel are engaged!' +num_episodes: 2427 +num_examples: 46610 diff --git a/parlai/tasks/friends/test/friends_valid.yml b/parlai/tasks/friends/test/friends_valid.yml new file mode 100644 index 00000000000..9c58364243b --- /dev/null +++ b/parlai/tasks/friends/test/friends_valid.yml @@ -0,0 +1,73 @@ +acts: +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: Chandler, if it really hurts that bad you should just tell + her.' + id: friends + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was!' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' + id: friends + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Ross Geller: Hey.' + id: friends + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' + id: friends + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey.' +- - characters: Chandler Bing,Joey Tribbiani,Monica Geller,Phoebe Buffay,Rachel Green,Ross + Geller + episode_done: false + eval_labels: + - 'Ross Geller: Sorry.' + id: friends + text: 'Chandler Bing: I''m telling you, she gives the worst massages ever!! Okay, + it was like she was torturing me for information. And I wanted to give it up + I just-I didn''t know what it was! + + Joey Tribbiani: Chandler, if it really hurts that bad you should just tell her. + + Chandler Bing: Look, for the first time in my life I''m in a real relationship. + Okay, I''m not gonna screw that up by y''know, telling the truth. + + Ross Geller: Hey. + + Joey Tribbiani: Whoa, dude, look out! You almost crushed my hat!' +num_episodes: 308 +num_examples: 5855 diff --git a/parlai/tasks/task_list.py b/parlai/tasks/task_list.py index acb32d1eacd..b72be9f7ea7 100644 --- a/parlai/tasks/task_list.py +++ b/parlai/tasks/task_list.py @@ -288,6 +288,17 @@ ), "links": {"arXiv": "https://arxiv.org/abs/1706.05125"}, }, + { + "id": "Friends", + "display_name": "Friends", + "task": "friends", + "tags": ["MultiPartyConvo"], + "description": ( + "Multi-party conversation dataset modified from the 10 seasons " + "of the popular American sitcom that ran in the 90s, Friends." + ), + "links": {"website": "https://convokit.cornell.edu/documentation/friends.html"}, + }, { "id": "Glue", "display_name": "Glue",