diff --git a/baselines/BERT/run_bert.py b/baselines/BERT/run_bert.py index 2f429cb9adb2751c65941fb54eb14749ac5c4d4e..da865181e6ad43b5f4165807d49e71b3a915364a 100644 --- a/baselines/BERT/run_bert.py +++ b/baselines/BERT/run_bert.py @@ -278,7 +278,7 @@ def train(train_set, dev_set, test_set, device, args): def load_data(args): logger.info("Creating data loaders") data_dir = '../../data' - with open(f'{data_dir}/data_split.json') as json_file: + with open(f'{data_dir}/data-split.json') as json_file: id_dict = json.load(json_file) if args.select_data == 'emowoz': diff --git a/baselines/COSMIC/feature-extraction/extract_bert_features.py b/baselines/COSMIC/feature-extraction/extract_bert_features.py index eee12ca3627eb6b2be9b0e27eeb598b1b42c1f3a..99c790941736812806554cb43f9d14726a9020ed 100644 --- a/baselines/COSMIC/feature-extraction/extract_bert_features.py +++ b/baselines/COSMIC/feature-extraction/extract_bert_features.py @@ -170,7 +170,7 @@ def extract(args, device): feature_dicts = get_sentence_encodings(model, emowoz) print("\nFormatting data") - with open(f'{data_dir}/data_split.json') as json_file: + with open(f'{data_dir}/data-split.json') as json_file: id_dict = json.load(json_file) train_ids = id_dict['train']['multiwoz'] + id_dict['train']['dialmage'] diff --git a/baselines/DialogueRNN/extract_bert_features.py b/baselines/DialogueRNN/extract_bert_features.py index 2bd7531f0a6c5c3064662e24829daebc1315e5b8..449080a7164c8ab2b9dd547c38aac6ff74444b0b 100644 --- a/baselines/DialogueRNN/extract_bert_features.py +++ b/baselines/DialogueRNN/extract_bert_features.py @@ -153,7 +153,7 @@ def extract(args, device): feature_dict = get_sentence_encodings(model, emowoz) print("\nFormatting data") - with open(f'{data_dir}/data_split.json') as json_file: + with open(f'{data_dir}/data-split.json') as json_file: id_dict = json.load(json_file) train_ids = id_dict['train']['multiwoz'] + id_dict['train']['dialmage'] diff --git a/baselines/DialogueRNN/preprocess_emowoz_for_glove.py b/baselines/DialogueRNN/preprocess_emowoz_for_glove.py index dc5b4e7924da981bc526e6962d4994d23d6ad9bd..62583b322e37ef46887d6b9fa2e9722a0ce406b7 100644 --- a/baselines/DialogueRNN/preprocess_emowoz_for_glove.py +++ b/baselines/DialogueRNN/preprocess_emowoz_for_glove.py @@ -6,7 +6,7 @@ from tqdm import tqdm print(f"Preprocessing emowoz for glove feature extraction.") data_dir = '../../data' -with open(f'{data_dir}/data_split.json') as json_file: +with open(f'{data_dir}/data-split.json') as json_file: id_dict = json.load(json_file) train_ids = id_dict['train']['multiwoz'] + id_dict['train']['dialmage']