Skip to content

Commit a4653a1

Browse files
knn bf16
1 parent fcaa907 commit a4653a1

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

configs/regular/bf16/knn.json

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,19 @@
1717
"algorithm": { "estimator_params": { "n_jobs": "[SPECIAL_VALUE]physical_cpus" } }
1818
},
1919
"synthetic classification data": {
20+
"algorithm": {
21+
"estimator": "KNeighborsClassifier",
22+
"estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": [1, 2] }
23+
},
2024
"data": [
2125
{ "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" } }
2226
]
2327
},
2428
"synthetic regression data": {
29+
"algorithm": {
30+
"estimator": "KNeighborsRegressor",
31+
"estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": [1, 2] }
32+
},
2533
"data": [
2634
{ "source": "make_regression", "split_kwargs": { "train_size": 50000, "test_size": 1000 }, "generation_kwargs": { "n_samples": 51000, "n_features": 100, "noise":1.5 } }
2735
]
@@ -33,7 +41,14 @@
3341
"sklearn-ex[gpu] implementations",
3442
"common knn parameters",
3543
"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",
3752
"synthetic regression data"
3853
]
3954
}

0 commit comments

Comments
 (0)