def get_data_handler_config(party_id, dataset, folder_data, is_agg=False): SUPPORTED_DATASETS = ['mnist'] if dataset in SUPPORTED_DATASETS: data = datahandlers.get_datahandler_config(dataset, folder_data, party_id, is_agg) else: raise Exception( "The dataset {} is a wrong combination for fusion/model".format( dataset)) return data
def get_data_handler_config(party_id, dataset, folder_data, is_agg=False): if is_agg: return None dataset = 'pendulum' data = datahandlers.get_datahandler_config(dataset, folder_data, party_id, is_agg) return data