Skip to content

Commit 4ac119e

Browse files
config fix
1 parent 1779a9f commit 4ac119e

File tree

1 file changed

+40
-4
lines changed

1 file changed

+40
-4
lines changed

configs/spmd/large_scale/spmd_for_online.json

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

0 commit comments

Comments
 (0)