(z: float)
| 130 | train_mask[i] = False |
| 131 | |
| 132 | train_idx = [i for i, keep in enumerate(train_mask) if keep] |
| 133 | if not train_idx or not test_idx: |
| 134 | raise ValueError("PurgedKFold generated an empty train/test fold") |
| 135 | |
| 136 | splits.append((train_idx, test_idx)) |
| 137 | start = stop |
| 138 | |
| 139 | return splits |
| 140 |