in E2E_TOD/reader.py [0:0]
def get_exp_domains(self, exp_domains, all_domains_list):
if 'hotel' in exp_domains:
if 'except' in exp_domains:
# ['except', 'hotel']
domains = [d for d in all_domains_list if 'hotel' not in d and 'multi' not in d]
else:
# ['hotel']
domains = ['hotel_single', 'hotel_multi']
if 'train' in exp_domains:
if 'except' in exp_domains:
# ['except', 'train']
domains = [d for d in all_domains_list if 'train' not in d and 'multi' not in d]
else:
# ['train']
domains = ['train_single', 'train_multi']
if 'attraction' in exp_domains:
if 'except' in exp_domains:
# ['except', 'attraction']
domains = [d for d in all_domains_list if 'attraction' not in d and 'multi' not in d]
else:
# ['attraction']
domains = ['attraction_single', 'attraction_multi']
if 'restaurant' in exp_domains:
if 'except' in exp_domains:
# ['except', 'restaurant']
domains = [d for d in all_domains_list if 'restaurant' not in d and 'multi' not in d]
else:
# ['restaurant']
domains = ['restaurant_single', 'restaurant_multi']
if 'taxi' in exp_domains:
if 'except' in exp_domains:
# ['except', 'taxi']
domains = [d for d in all_domains_list if 'taxi' not in d and 'multi' not in d]
else:
# ['taxi']
domains = ['taxi_single', 'taxi_multi']
return domains