|
17 | 17 | "algorithm": { "estimator_params": { "n_jobs": "[SPECIAL_VALUE]physical_cpus" } }
|
18 | 18 | },
|
19 | 19 | "synthetic classification data": {
|
| 20 | + "algorithm": { |
| 21 | + "estimator": "KNeighborsClassifier", |
| 22 | + "estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": [1, 2] } |
| 23 | + }, |
20 | 24 | "data": [
|
21 | 25 | { "source": "make_classification", "split_kwargs": { "train_size": 50000, "test_size": 1000 }, "generation_kwargs": { "n_samples": 51000, "n_features": 100, "n_classes": 2, "n_informative": "[SPECIAL_VALUE]0.5" } }
|
22 | 26 | ]
|
23 | 27 | },
|
24 | 28 | "synthetic regression data": {
|
| 29 | + "algorithm": { |
| 30 | + "estimator": "KNeighborsRegressor", |
| 31 | + "estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": [1, 2] } |
| 32 | + }, |
25 | 33 | "data": [
|
26 | 34 | { "source": "make_regression", "split_kwargs": { "train_size": 50000, "test_size": 1000 }, "generation_kwargs": { "n_samples": 51000, "n_features": 100, "noise":1.5 } }
|
27 | 35 | ]
|
|
33 | 41 | "sklearn-ex[gpu] implementations",
|
34 | 42 | "common knn parameters",
|
35 | 43 | "sklearn knn parameters",
|
36 |
| - "synthetic classification data", |
| 44 | + "synthetic classification data" |
| 45 | + ] |
| 46 | + }, |
| 47 | + "sklearn brute knn reg": { |
| 48 | + "SETS": [ |
| 49 | + "sklearn-ex[gpu] implementations", |
| 50 | + "common knn parameters", |
| 51 | + "sklearn knn parameters", |
37 | 52 | "synthetic regression data"
|
38 | 53 | ]
|
39 | 54 | }
|
|
0 commit comments