Skip to content

Commit 687178b

Browse files
incremental spmd
1 parent c47649a commit 687178b

File tree

4 files changed

+121
-0
lines changed

4 files changed

+121
-0
lines changed
Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
{
2+
"INCLUDE": ["../../../common/sklearn.json", "../../../spmd/stats_covariance.json", "../large_scale.json"],
3+
"PARAMETERS_SETS": {
4+
"spmd basicstats parameters": {
5+
"algorithm": {
6+
"estimator": "IncrementalBasicStatistics",
7+
"estimator_methods": { "training": "fit" },
8+
"num_batches": {"training": 10}
9+
},
10+
"data": {
11+
"split_kwargs": { "test_size": 0.0001 }
12+
}
13+
},
14+
"synthetic data": {
15+
"data": [
16+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 10000000, "n_features": 10, "centers": 1 } },
17+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 100000, "n_features": 1000, "centers": 1 } }
18+
]
19+
}
20+
},
21+
"TEMPLATES": {
22+
"basicstats": {
23+
"SETS": [
24+
"sklearnex spmd implementation",
25+
"large scale 64 parameters",
26+
"synthetic data",
27+
"spmd basicstats parameters"
28+
]
29+
}
30+
}
31+
}
Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
{
2+
"INCLUDE": ["../../../common/sklearn.json", "../../../spmd/stats_covariance.json", "../large_scale.json"],
3+
"PARAMETERS_SETS": {
4+
"spmd covariance parameters": {
5+
"algorithm": {
6+
"estimator": "IncrementalEmpiricalCovariance",
7+
"estimator_methods": { "training": "fit" },
8+
"num_batches": {"training": 10}
9+
},
10+
"data": {
11+
"split_kwargs": { "test_size": 0.0001 }
12+
}
13+
},
14+
"synthetic data": {
15+
"data": [
16+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 10000000, "n_features": 10, "centers": 1 } },
17+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 100000, "n_features": 1000, "centers": 1 } }
18+
]
19+
}
20+
},
21+
"TEMPLATES": {
22+
"covariance": {
23+
"SETS": [
24+
"sklearnex spmd implementation",
25+
"large scale 64 parameters",
26+
"synthetic data",
27+
"spmd covariance parameters"
28+
]
29+
}
30+
}
31+
}
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
{
2+
"INCLUDE": ["../../../common/sklearn.json", "../../../regular/linear_model.json", "../large_scale.json"],
3+
"PARAMETERS_SETS": {
4+
"spmd linear parameters": {
5+
"algorithm": {
6+
"estimator": "IncrementalLinearRegression",
7+
"estimator_methods": { "training": "fit" },
8+
"num_batches": {"training": 10}
9+
}
10+
},
11+
"synthetic data": {
12+
"data": [
13+
{ "source": "make_regression", "generation_kwargs": { "n_samples": 30005000, "n_features": 10, "noise": 1.25 }, "split_kwargs": { "train_size": 30000000, "test_size": 5000 } },
14+
{ "source": "make_regression", "generation_kwargs": { "n_samples": 305000, "n_features": 1000, "noise": 1.25 }, "split_kwargs": { "train_size": 300000, "test_size": 5000 } }
15+
]
16+
}
17+
},
18+
"TEMPLATES": {
19+
"linreg": {
20+
"SETS": [
21+
"sklearnex spmd implementation",
22+
"large scale 64 parameters",
23+
"synthetic data",
24+
"spmd linear parameters"
25+
]
26+
}
27+
}
28+
}
Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
{
2+
"INCLUDE": ["../../../common/sklearn.json", "../../../regular/pca.json", "../large_scale.json"],
3+
"PARAMETERS_SETS": {
4+
"spmd pca parameters": {
5+
"algorithm": {
6+
"estimator": "IncrementalPCA",
7+
"estimator_methods": { "training": "fit", "inference": "" },
8+
"num_batches": {"training": 10}
9+
},
10+
"data": {
11+
"split_kwargs": { "test_size": 0.0001 }
12+
}
13+
},
14+
"synthetic data": {
15+
"data": [
16+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 10000000, "n_features": 10, "centers": 1 } },
17+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 100000, "n_features": 1000, "centers": 1 } }
18+
]
19+
}
20+
},
21+
"TEMPLATES": {
22+
"linreg": {
23+
"SETS": [
24+
"sklearnex spmd implementation",
25+
"large scale 64 parameters",
26+
"synthetic data",
27+
"spmd pca parameters"
28+
]
29+
}
30+
}
31+
}

0 commit comments

Comments
 (0)