def prepare_data(source_lang, target_lang, reverse=False):
filename = '%s-%s.txt' % (source_lang, target_lang)
pairs = get_pair(os.path.join(data_root, filename), reverse)
if reverse:
source_lang_dict = Lang(target_lang)
target_lang_dict = Lang(source_lang)
else:
source_lang_dict = Lang(source_lang)
target_lang_dict = Lang(target_lang)
for p in pairs:
source_lang_dict.add_word_list(p[0])
target_lang_dict.add_word_list(p[1])
return pairs, source_lang_dict, target_lang_dict
