Change benchmarks configuration

kdtree/benchmarks/bench.py

@@ -27,7 +27,7 @@ def benchmark(config, results_folder, bench_name):
     datasets = config["datasets"]
     estimators = config["estimators"]
     leaf_sizes = config["leaf_sizes"]
-    n_neighbors = config["n_neighbors"]
+    n_neighbors = config.get("n_neighbors", [])
 
     n_trials = config.get("n_trials", 3)
     return_distance = config.get("return_distance", False)

kdtree/benchmarks/config.yml

@@ -4,18 +4,21 @@ estimators:
   - name: cython+
     estimator: kdtree.KDTree
 
-n_trials: 1
+n_trials: 3
 
 datasets:
   - n_samples_train: 1e6
-    n_samples_test: 1e4
+    n_samples_test: 1e6
     n_features: 5
   - n_samples_train: 1e6
-    n_samples_test: 1e4
+    n_samples_test: 1e6
     n_features: 10
   - n_samples_train: 1e6
-    n_samples_test: 1e4
+    n_samples_test: 1e6
     n_features: 50
+  - n_samples_train: 1e6
+    n_samples_test: 1e6
+    n_features: 100
 
 leaf_sizes:
   - 64
@@ -24,8 +27,4 @@ leaf_sizes:
   - 512
   - 1024
   - 2048
-
-n_neighbors:
- - 1
- - 10
- - 100
\ No newline at end of file
+  - 4096