Total Complexity | 1 |
Total Lines | 8 |
Duplicated Lines | 0 % |
1 | from fuel.datasets import H5PYDataset |
||
5 | class Adult(H5PYDataset): |
||
6 | filename = 'adult.hdf5' |
||
7 | |||
8 | def __init__(self, which_sets, **kwargs): |
||
9 | kwargs.setdefault('load_in_memory', True) |
||
10 | super(Adult, self).__init__( |
||
11 | file_or_path=find_in_data_path(self.filename), |
||
12 | which_sets=which_sets, **kwargs |
||
13 | ) |
||
14 |