Yield next batch of data
(self, input_data: Callable)
| 88 | self._iter = 0 |
| 89 | |
| 90 | def next(self, input_data: Callable) -> bool: |
| 91 | """Yield next batch of data""" |
| 92 | if self._iter == len(self._data): |
| 93 | # Return False when there's no more batch. |
| 94 | return False |
| 95 | |
| 96 | kwargs = {k: self._get(k) for k in meta} |
| 97 | input_data( |
| 98 | data=self.data(), |
| 99 | group=None, |
| 100 | feature_names=self._feature_names, |
| 101 | feature_types=self._feature_types, |
| 102 | feature_weights=self._feature_weights, |
| 103 | **kwargs, |
| 104 | ) |
| 105 | self._iter += 1 |
| 106 | return True |
| 107 | |
| 108 | |
| 109 | @overload |