zomux /
deepy
| @@ 55-68 (lines=14) @@ | ||
| 52 | else: |
|
| 53 | return data_generator |
|
| 54 | ||
| 55 | def valid_set(self): |
|
| 56 | if not self.origin.valid_set(): |
|
| 57 | return None |
|
| 58 | if self.cache and self._cached_valid_set is not None: |
|
| 59 | return self._cached_valid_set |
|
| 60 | ||
| 61 | data_generator = self._yield_data(self.origin.valid_set()) |
|
| 62 | if data_generator is None: |
|
| 63 | return None |
|
| 64 | if self.cache: |
|
| 65 | self._cached_valid_set = list(data_generator) |
|
| 66 | return self._cached_valid_set |
|
| 67 | else: |
|
| 68 | return data_generator |
|
| 69 | ||
| 70 | def train_size(self): |
|
| 71 | train_size = self.origin.train_size() |
|
| @@ 40-53 (lines=14) @@ | ||
| 37 | else: |
|
| 38 | return data_generator |
|
| 39 | ||
| 40 | def test_set(self): |
|
| 41 | if not self.origin.test_set(): |
|
| 42 | return None |
|
| 43 | if self.cache and self._cached_test_set is not None: |
|
| 44 | return self._cached_test_set |
|
| 45 | ||
| 46 | data_generator = self._yield_data(self.origin.test_set()) |
|
| 47 | if data_generator is None: |
|
| 48 | return None |
|
| 49 | if self.cache: |
|
| 50 | self._cached_test_set = list(data_generator) |
|
| 51 | return self._cached_test_set |
|
| 52 | else: |
|
| 53 | return data_generator |
|
| 54 | ||
| 55 | def valid_set(self): |
|
| 56 | if not self.origin.valid_set(): |
|
