Generate a batch of vectors
(
base_idx: int, batch_size: int, dimensions: int, perturbation_degree: float
)
| 70 | |
| 71 | |
| 72 | def generate_batch( |
| 73 | base_idx: int, batch_size: int, dimensions: int, perturbation_degree: float |
| 74 | ) -> list: |
| 75 | """Generate a batch of vectors""" |
| 76 | batch_vectors = [] |
| 77 | base_vector = generate_random_vector(base_idx * batch_size, dimensions) |
| 78 | batch_vectors.append(base_vector) |
| 79 | |
| 80 | for i in range(batch_size - 1): |
| 81 | perturbed_vector = generate_perturbation( |
| 82 | base_vector, base_idx * batch_size + i + 1, perturbation_degree |
| 83 | ) |
| 84 | batch_vectors.append(perturbed_vector) |
| 85 | |
| 86 | return batch_vectors |
| 87 | |
| 88 | |
| 89 | def test_performance(): |
nothing calls this directly
no test coverage detected