Exemplos de get_datahandler_config em Python

Exemplo n.º 1

Arquivo: generate_configs.py Projeto: Fizzzu/Pneumonia_Classification_

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

Exemplo n.º 2

Arquivo: generate_configs.py Projeto: kedar-devs/IBM-FL-PNEUMONIA-Classification

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