diff --git a/benchmarks/index_creation.py b/benchmarks/index_creation.py index 1a2f587e..95b227d4 100644 --- a/benchmarks/index_creation.py +++ b/benchmarks/index_creation.py @@ -24,7 +24,7 @@ class IndexCreationSuite: params = ( [256], [1024], - [voyager.Space.Euclidean, voyager.Space.Cosine], + [voyager.Space.Euclidean, voyager.Space.InnerProduct, voyager.Space.Cosine], [voyager.StorageDataType.E4M3, voyager.StorageDataType.Float8, voyager.StorageDataType.Float32], [24], ) @@ -50,6 +50,7 @@ def setup( ef_construction=ef_construction, M=20, storage_data_type=storage_data_type, + random_seed=4321, ) self.input_data = input_data diff --git a/benchmarks/index_query.py b/benchmarks/index_query.py index b3865f11..e3c9be03 100644 --- a/benchmarks/index_query.py +++ b/benchmarks/index_query.py @@ -27,8 +27,8 @@ class IndexQuerySuite: repeat = (1, 10, 30.0) params = ( [256], - [1024], - [voyager.Space.Euclidean, voyager.Space.Cosine], + [4096], + [voyager.Space.Euclidean, voyager.Space.InnerProduct, voyager.Space.Cosine], [voyager.StorageDataType.E4M3, voyager.StorageDataType.Float8, voyager.StorageDataType.Float32], [24], ) @@ -54,8 +54,9 @@ def setup_cache(self) -> Dict: ef_construction=ef_construction, M=20, storage_data_type=storage_data_type, + random_seed=4321, ) - index.add_items(input_data) + index.add_items(input_data, num_threads=1) data[param_combination] = (index.as_bytes(), input_data)