|
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