def load_dataset(dataset_yml, use_test_set):
    """
    Load the dataset used by the model.

    Parameters
    ----------
    dataset_yml: str
        Yaml description of the dataset.
    """

    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