Select a number of records from the dataset.
(self, n, random=False)
| 45 | self.transforms = transforms |
| 46 | |
| 47 | def select(self, n, random=False): |
| 48 | """ |
| 49 | Select a number of records from the dataset. |
| 50 | """ |
| 51 | all_indices = np.arange(len(self)) |
| 52 | if random: |
| 53 | selected_indices = np.random.choice(all_indices, size=n, replace=False) |
| 54 | else: |
| 55 | selected_indices = np.arange(n) |
| 56 | |
| 57 | ds = deepcopy(self) |
| 58 | ds.inputs = ds.inputs.select(selected_indices) |
| 59 | ds.outputs = ds.outputs.select(selected_indices) |
| 60 | |
| 61 | return ds |
| 62 | |
| 63 | def split(self, p, random=False): |
| 64 | """ |