Skip to content

Commit fce0651

Browse files
KNN strong
1 parent 4fee991 commit fce0651

File tree

1 file changed

+38
-0
lines changed

1 file changed

+38
-0
lines changed
Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
{
2+
"INCLUDE": ["../../common/sklearn.json", "../../regular/knn.json", "large_scale.json"],
3+
"PARAMETERS_SETS": {
4+
"spmd knn cls parameters": {
5+
"algorithm": {
6+
"estimator": "KNeighborsClassifier",
7+
"estimator_params": {
8+
"algorithm": "brute",
9+
"metric": "minkowski",
10+
"p": 2,
11+
"weights": "uniform",
12+
"n_neighbors": 5
13+
},
14+
"estimator_methods": {
15+
"training": "fit",
16+
"inference": "predict"
17+
}
18+
}
19+
},
20+
"synthetic classification data": {
21+
"data": [
22+
{ "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" } },
23+
{ "source": "make_classification", "split_kwargs": { "train_size": 50000, "test_size": 500000 }, "generation_kwargs": { "n_samples": 550000, "n_features": 100, "n_classes": 2, "n_informative": "[SPECIAL_VALUE]0.5" } }
24+
]
25+
}
26+
},
27+
"TEMPLATES": {
28+
"knn classifier": {
29+
"SETS": [
30+
"common knn parameters",
31+
"synthetic classification data",
32+
"sklearnex spmd implementation",
33+
"large scale strong 2k parameters",
34+
"spmd knn cls parameters"
35+
]
36+
}
37+
}
38+
}

0 commit comments

Comments
 (0)