def get_model_selection(dataset_name):
    """ Returns the model selection for a dataset

    Args:
        dataset_name (str): Name of the dataset

    Returns:
        list: list of model selection name 
    """

    if dataset_name in [ 'Spurious_Fourier', "TCMNIST_Source", "TCMNIST_Time"]:
        return ['train_domain_validation', 'test_domain_validation']
    if dataset_name in [ 'CAP', 'SEDFx', 'PCL', 'LSA64', 'HHAR']:
        return ['train_domain_validation', 'oracle_train_domain_validation']
    if dataset_name in ['AusElectricity', 'AusElectricityUnbalanced', 'IEMOCAPUnbalanced', 'IEMOCAP']:
        return ['average_validation', 'weighted_average_validation', 'worse_domain_validation']