

def load_dataset(dataset_yml, use_test_set):
    '\n    Load the dataset used by the model.\n\n    Parameters\n    ----------\n    dataset_yml: str\n        Yaml description of the dataset.\n    '
    print('Loading data...')
    dataset = yaml_parse.load(dataset_yml)
    if (use_test_set == 'y'):
        dataset = dataset.get_test_set()
    else:
        assert (use_test_set == 'n')
    return dataset

dataset/ETHPy150Open lisa-lab/pylearn2/pylearn2/scripts/dbm/show_reconstructions.py load_dataset/original