Skip to content

Commit d172d2a

Browse files
knn bf16
1 parent 270c841 commit d172d2a

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

configs/regular/bf16/knn.json

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,17 +17,23 @@
1717
},
1818
"synthetic classification data": {
1919
"data": [
20-
{ "source": "make_classification", "split_kwargs": { "train_size": 500000, "test_size": 50000 }, "generation_kwargs": { "n_samples": 550000, "n_features": 100, "n_classes": 2, "n_informative": "[SPECIAL_VALUE]0.5" } }
20+
{ "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" } }
2121
]
22-
}
22+
},
23+
"synthetic regression data": {
24+
"data": [
25+
{ "source": "make_regression", "split_kwargs": { "train_size": 50000, "test_size": 1000 }, "generation_kwargs": { "n_samples": 51000, "n_features": 100, "noise":1.5 } }
26+
]
27+
}
2328
},
2429
"TEMPLATES": {
2530
"sklearn brute knn clsf": {
2631
"SETS": [
2732
"sklearn-ex[gpu] implementations",
2833
"common knn parameters",
2934
"sklearn knn parameters",
30-
"synthetic classification data"
35+
"synthetic classification data",
36+
"synthetic regression data"
3137
]
3238
}
3339
}

0 commit comments

Comments
 (0)