Include leaf_size in benchmark

kdtree/benchmarks/benchmark.py

@@ -65,7 +65,7 @@ def benchmark(config, results_folder, bench_name):
                 bytes_processed_data_query = X_test.nbytes
 
                 t0_ = time.perf_counter()
-                sk_tree = KDTree(X_train, leaf_size=256)
+                sk_tree = KDTree(X_train, leaf_size=leaf_size)
                 t1_ = time.perf_counter()
                 time_elapsed = round(t1_ - t0_, 5)
 
@@ -87,7 +87,7 @@ def benchmark(config, results_folder, bench_name):
                 print("---")
 
                 t0_ = time.perf_counter()
-                tree = kdtree.KDTree(X_train, leaf_size=256)
+                tree = kdtree.KDTree(X_train, leaf_size=leaf_size)
                 t1_ = time.perf_counter()
                 time_elapsed = round(t1_ - t0_, 5)
 
@@ -184,6 +184,7 @@ def report(results_folder, bench_name):
     cols = [
         "n_samples_train",
         "n_features",
+        "leaf_size",
     ]
 
     df[cols] = df[cols].astype(np.uint32)
@@ -220,7 +221,7 @@ def report(results_folder, bench_name):
             f"Euclidean Distance, dtype=np.float64, {df.trial.max() + 1} trials - Bench. Name: {bench_name}\n"
         )
         title += (
-            "n_samples_train=%s - n_features=%s" % vals
+            "n_samples_train=%s - n_features=%s - leaf_size=%s" % vals
         )
         _ = fig.suptitle(title, fontsize=16)
         plt.savefig(f"{results_folder}/{bench_name}_{i}.pdf", bbox_inches="tight")