A very little helper class for randomly pick data samples from your dataset.
| 126 | |
| 127 | |
| 128 | class RandomMemDataset: |
| 129 | """A very little helper class for randomly pick data samples from your |
| 130 | dataset.""" |
| 131 | def __init__(self, data: Iterable) -> None: |
| 132 | self._data = data |
| 133 | self._num_of_batchs = len(data) |
| 134 | |
| 135 | def pop(self): |
| 136 | idx = random.randint(0, self._num_of_batchs - 1) |
| 137 | return self._data[idx] |
| 138 | |
| 139 | |
| 140 | class PriorityQueue: |