44// Tests all three deglib::builder::OptimizationTarget modes:
55// - LowLID, HighLID, StreamingData
66
7- TEST (DeglibBuilderRegressionL2, OptimizationTargetsBenchmark )
7+ TEST (DeglibBuilderRegressionL2, Benchmark_LowLID )
88{
99 const size_t dim = 128 ;
1010 const size_t base_count = 100000 ;
@@ -17,17 +17,45 @@ TEST(DeglibBuilderRegressionL2, OptimizationTargetsBenchmark)
1717
1818 auto gt_data = compute_groundtruth_l2 (base_data, base_count, query_data, query_count, dim, 10 );
1919
20- run_regression_test (" LowLID" , deglib::Metric::L2 , 26000 .0 , 10.0 , 0.959 ,
20+ run_regression_test (" LowLID" , deglib::Metric::L2 , 30000 .0 , 9.3 , 0.92 ,
2121 base_data.data (), query_data.data (), base_count, query_count, dim, gt_data,
2222 deglib::distances::fp32_l2::L2Float{}, 100 ,
2323 deglib::builder::OptimizationTarget::LowLID);
24+ }
25+
26+ TEST (DeglibBuilderRegressionL2, Benchmark_HighLID)
27+ {
28+ const size_t dim = 128 ;
29+ const size_t base_count = 100000 ;
30+ const size_t query_count = 100 ;
31+ const size_t num_clusters = 1000 ;
32+
33+ std::vector<float > base_data;
34+ std::vector<float > query_data;
35+ generate_synthetic_clustered_dataset (base_count, dim, base_data, query_data, query_count, num_clusters);
36+
37+ auto gt_data = compute_groundtruth_l2 (base_data, base_count, query_data, query_count, dim, 10 );
2438
25- run_regression_test (" HighLID" , deglib::Metric::L2 , 15000.0 , 10.0 , 0.888 ,
39+ run_regression_test (" HighLID" , deglib::Metric::L2 , 15000.0 , 10.0 , 0.864 ,
2640 base_data.data (), query_data.data (), base_count, query_count, dim, gt_data,
2741 deglib::distances::fp32_l2::L2Float{}, 100 ,
2842 deglib::builder::OptimizationTarget::HighLID);
43+ }
44+
45+ TEST (DeglibBuilderRegressionL2, Benchmark_StreamingData)
46+ {
47+ const size_t dim = 128 ;
48+ const size_t base_count = 100000 ;
49+ const size_t query_count = 100 ;
50+ const size_t num_clusters = 1000 ;
51+
52+ std::vector<float > base_data;
53+ std::vector<float > query_data;
54+ generate_synthetic_clustered_dataset (base_count, dim, base_data, query_data, query_count, num_clusters);
55+
56+ auto gt_data = compute_groundtruth_l2 (base_data, base_count, query_data, query_count, dim, 10 );
2957
30- run_regression_test (" StreamingData" , deglib::Metric::L2 , 24000 .0 , 24.0 , 0.96 ,
58+ run_regression_test (" StreamingData" , deglib::Metric::L2 , 25000 .0 , 23.8 , 0.95 ,
3159 base_data.data (), query_data.data (), base_count, query_count, dim, gt_data,
3260 deglib::distances::fp32_l2::L2Float{}, 100 ,
3361 deglib::builder::OptimizationTarget::StreamingData);
0 commit comments