Skip to content

Commit a137f76

Browse files
committed
add preprocessing benchmarking
1 parent eb5333a commit a137f76

1 file changed

Lines changed: 41 additions & 0 deletions

File tree

‎configs/regular/preprocessing.json‎

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
{
2+
"INCLUDE": ["../common/sklearn.json"],
3+
"PARAMETERS_SETS": {
4+
"preprocessing parameters": {
5+
"algorithm": {
6+
"estimator": "MaxAbsScaler",
7+
"estimator_methods": { "training": "fit", "inference": "transform" }
8+
}
9+
},
10+
"preprocessing datasets": [
11+
{
12+
"data": {
13+
"source": "make_regression",
14+
"generation_kwargs": [
15+
{ "n_samples": 100000, "n_features": 5000, "noise": 1.0 },
16+
{ "n_samples": 1000000, "n_features": 500, "noise": 1.0 },
17+
{ "n_samples": 10000000, "n_features": 50, "noise": 1.0 }
18+
],
19+
"split_kwargs": { "train_size": 0.5, "test_size": 0.5 },
20+
"dtype": ["float32", "float64"]
21+
}
22+
},
23+
{
24+
"data": {
25+
"dataset": ["sift", "gist", "svhn"],
26+
"split_kwargs": { "train_size": 0.5, "test_size": 0.5 },
27+
"dtype": ["float32", "float64"]
28+
}
29+
}
30+
]
31+
},
32+
"TEMPLATES": {
33+
"sklearn preprocessing": {
34+
"SETS": [
35+
"sklearn-ex[cpu,gpu] implementations",
36+
"preprocessing parameters",
37+
"preprocessing datasets"
38+
]
39+
}
40+
}
41+
}

0 commit comments

Comments
 (0)