Remove small leaf sizes from benchmarks

63a742b6 · Julien Jerphanion · 76afeae8 · 63a742b6
Commit 63a742b6 authored Jul 16, 2021 by Julien Jerphanion
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 3 deletions

kdtree/benchmarks/config.yml kdtree/benchmarks/config.yml +10 -3

No files found.
--- a/kdtree/benchmarks/config.yml
+++ b/kdtree/benchmarks/config.yml
@@ -4,12 +4,18 @@ estimators:
  - name: cython+
    estimator: kdtree.KDTree

-n_trials: 10
+n_trials: 5

 datasets:
  - n_samples_train: 1e5
-    n_samples_test: 1e6
-    n_features: 50
+    n_samples_test: 1 # not used yet
+    n_features: 2
+  - n_samples_train: 1e5
+    n_samples_test: 1 # not used yet
+    n_features: 10
+  - n_samples_train: 1e5
+    n_samples_test: 1 # not used yet
+    n_features: 100

 leaf_sizes:
  - 64
@@ -19,3 +25,4 @@ leaf_sizes:
  - 1024
  - 2048
  - 4096
+  - 8192