|
4 | 4 | "common knn parameters": { |
5 | 5 | "algorithm": { |
6 | 6 | "estimator_params": { |
7 | | - "n_neighbors": [10, 100], |
| 7 | + "n_neighbors": 10, |
8 | 8 | "weights": "uniform" |
9 | 9 | } |
10 | 10 | }, |
|
19 | 19 | "synthetic classification data": { |
20 | 20 | "algorithm": { |
21 | 21 | "estimator": "KNeighborsClassifier", |
22 | | - "estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": [1, 2] } |
| 22 | + "estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": 2 } |
23 | 23 | }, |
24 | 24 | "data": [ |
25 | 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" } } |
26 | 26 | ] |
27 | | - }, |
28 | | - "synthetic regression data": { |
29 | | - "algorithm": { |
30 | | - "estimator": "KNeighborsRegressor", |
31 | | - "estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": [1, 2] } |
32 | | - }, |
33 | | - "data": [ |
34 | | - { "source": "make_regression", "split_kwargs": { "train_size": 5000000, "test_size": 1000 }, "generation_kwargs": { "n_samples": 5001000, "n_features": 100, "noise":1.5 } } |
35 | | - ] |
36 | 27 | } |
37 | 28 | }, |
38 | 29 | "TEMPLATES": { |
|
43 | 34 | "sklearn knn parameters", |
44 | 35 | "synthetic classification data" |
45 | 36 | ] |
46 | | - }, |
47 | | - "sklearn brute knn reg": { |
48 | | - "SETS": [ |
49 | | - "sklearn-ex[gpu] implementations", |
50 | | - "common knn parameters", |
51 | | - "sklearn knn parameters", |
52 | | - "synthetic regression data" |
53 | | - ] |
54 | 37 | } |
55 | 38 | } |
56 | 39 | } |
0 commit comments