Yields: Batched data by stacking each component on an extra 0th dimension.
(self)
| 112 | return div + int(self.remainder) |
| 113 | |
| 114 | def __iter__(self): |
| 115 | """ |
| 116 | Yields: |
| 117 | Batched data by stacking each component on an extra 0th dimension. |
| 118 | """ |
| 119 | holder = [] |
| 120 | for data in self.ds: |
| 121 | holder.append(data) |
| 122 | if len(holder) == self.batch_size: |
| 123 | yield BatchData.aggregate_batch(holder, self.use_list) |
| 124 | del holder[:] |
| 125 | if self.remainder and len(holder) > 0: |
| 126 | yield BatchData.aggregate_batch(holder, self.use_list) |
| 127 | |
| 128 | @staticmethod |
| 129 | def _batch_numpy(data_list): |
no test coverage detected