Module datagen

Data generation utilities for unit tests

Structs

BatchGenerator

IncrementingInt32

RandomVector

Traits

ArrayGenerator

Functions

generate_random_array

Create a random float32 array where each element is uniformly distributed between [0..1]

generate_random_array_with_range

Create a random primitive array where each element is uniformly distributed a given range.

generate_random_array_with_seed

Create a random float32 array.

generate_scaled_random_array

Create a random float32 array where each element is uniformly distributed across the given range

sample_indices

sample_without_replacement

some_batch

Returns a non-empty batch of data

some_indexable_batch

Returns a batch of data that has a column that can be used to create an ANN index