Skip to content

Commit 4b2f189

Browse files
committed
more knn alignment bf16
1 parent 0e2cac6 commit 4b2f189

File tree

1 file changed

+2
-19
lines changed

1 file changed

+2
-19
lines changed

configs/regular/bf16/knn.json

Lines changed: 2 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
"common knn parameters": {
55
"algorithm": {
66
"estimator_params": {
7-
"n_neighbors": [10, 100],
7+
"n_neighbors": 10,
88
"weights": "uniform"
99
}
1010
},
@@ -19,20 +19,11 @@
1919
"synthetic classification data": {
2020
"algorithm": {
2121
"estimator": "KNeighborsClassifier",
22-
"estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": [1, 2] }
22+
"estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": 2 }
2323
},
2424
"data": [
2525
{ "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" } }
2626
]
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-
]
3627
}
3728
},
3829
"TEMPLATES": {
@@ -43,14 +34,6 @@
4334
"sklearn knn parameters",
4435
"synthetic classification data"
4536
]
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-
]
5437
}
5538
}
5639
}

0 commit comments

Comments
 (0)