Skip to content

Commit 2377a9e

Browse files
spmd vs online fix
1 parent da7f425 commit 2377a9e

File tree

2 files changed

+10
-82
lines changed

2 files changed

+10
-82
lines changed

configs/spmd/large_scale/spmd_for_online.json

Lines changed: 5 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -47,50 +47,14 @@
4747
},
4848
"split_kwargs": {"ignore": true}
4949
}
50-
},
51-
"covariance": {
52-
"algorithm": [
53-
{
54-
"estimator": "EmpiricalCovariance",
55-
"library": "sklearnex.covariance",
56-
"estimator_methods": {"training": "fit"}
57-
}
58-
]
59-
},
60-
"basic_statistics": {
61-
"algorithm": [
62-
{
63-
"estimator": "BasicStatistics",
64-
"library": "sklearnex.basic_statistics",
65-
"estimator_methods": {"training": "fit"}
66-
}
67-
]
68-
},
69-
"linear_regression": {
70-
"algorithm": [
71-
{
72-
"estimator": "LinearRegression",
73-
"library": "sklearnex.linear_model",
74-
"estimator_methods": {"training": "fit"}
75-
}
76-
]
77-
},
78-
"pca": {
79-
"algorithm": [
80-
{
81-
"estimator": "IncrementalPCA",
82-
"library": "sklearnex.decomposition",
83-
"estimator_methods": {"training": "fit"}
84-
}
85-
]
8650
}
8751
},
8852
"TEMPLATES": {
89-
"basic_statistics": {"SETS": ["basic_statistics", "basic_statistics data", "large scale strong full one node parameters", "sklearnex spmd implementation"]},
90-
"covariance": {"SETS": ["covariance", "covariance data", "large scale strong full one node parameters", "sklearnex spmd implementation"]},
53+
"basic_statistics": {"SETS": ["basic_statistics data", "spmd default parameters", "sklearnex spmd implementation", "large scale full one node parameters"]},
54+
"covariance": {"SETS": ["covariance data", "spmd default parameters","sklearnex spmd implementation", "large scale full one node parameters"]},
9155
"linear_regression": {
92-
"SETS": ["linear_regression", "linear_regression data", "large scale strong full one node parameters", "sklearnex spmd implementation"]
56+
"SETS": ["linear_regression data", "spmd default parameters", "sklearnex spmd implementation", "large scale full one node parameters"]
9357
},
94-
"pca": {"SETS": ["pca", "pca data", "large scale strong full one node parameters", "sklearnex spmd implementation"]}
58+
"pca": {"SETS": ["pca data", "spmd default parameters", "sklearnex spmd implementation", "large scale full one node parameters"]}
9559
}
96-
}
60+
}

configs/spmd/large_scale/spmd_for_online_strong.json

Lines changed: 5 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -47,50 +47,14 @@
4747
},
4848
"split_kwargs": {"ignore": true}
4949
}
50-
},
51-
"covariance": {
52-
"algorithm": [
53-
{
54-
"estimator": "EmpiricalCovariance",
55-
"library": "sklearnex.covariance",
56-
"estimator_methods": {"training": "fit"}
57-
}
58-
]
59-
},
60-
"basic_statistics": {
61-
"algorithm": [
62-
{
63-
"estimator": "BasicStatistics",
64-
"library": "sklearnex.basic_statistics",
65-
"estimator_methods": {"training": "fit"}
66-
}
67-
]
68-
},
69-
"linear_regression": {
70-
"algorithm": [
71-
{
72-
"estimator": "LinearRegression",
73-
"library": "sklearnex.linear_model",
74-
"estimator_methods": {"training": "fit"}
75-
}
76-
]
77-
},
78-
"pca": {
79-
"algorithm": [
80-
{
81-
"estimator": "IncrementalPCA",
82-
"library": "sklearnex.decomposition",
83-
"estimator_methods": {"training": "fit"}
84-
}
85-
]
8650
}
8751
},
8852
"TEMPLATES": {
89-
"basic_statistics": {"SETS": ["basic_statistics", "basic_statistics data", "large scale strong full one node parameters", "sklearnex spmd implementation"]},
90-
"covariance": {"SETS": ["covariance", "covariance data", "large scale strong full one node parameters", "sklearnex spmd implementation"]},
53+
"basic_statistics": {"SETS": ["basic_statistics data", "spmd default parameters", "sklearnex spmd implementation", "large scale strong full one node parameters"]},
54+
"covariance": {"SETS": ["covariance data", "spmd default parameters","sklearnex spmd implementation", "large scale strong full one node parameters"]},
9155
"linear_regression": {
92-
"SETS": ["linear_regression", "linear_regression data", "large scale strong full one node parameters", "sklearnex spmd implementation"]
56+
"SETS": ["linear_regression data", "spmd default parameters", "sklearnex spmd implementation", "large scale strong full one node parameters"]
9357
},
94-
"pca": {"SETS": ["pca", "pca data", "large scale strong full one node parameters", "sklearnex spmd implementation"]}
58+
"pca": {"SETS": ["pca data", "spmd default parameters", "sklearnex spmd implementation", "large scale strong full one node parameters"]}
9559
}
96-
}
60+
}

0 commit comments

Comments
 (0)