Skip to content

Commit d44dad9

Browse files
ethanglaserethanglaser
committed
minor update for kmeans out of preview
1 parent e76e463 commit d44dad9

File tree

2 files changed

+246
-1
lines changed

2 files changed

+246
-1
lines changed

‎configs/regular/kmeans.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@
7070
"TEMPLATES": {
7171
"sklearn kmeans": {
7272
"SETS": [
73-
"sklearn-ex[preview] implementations",
73+
"sklearn-ex[cpu,gpu] implementations",
7474
"common kmeans parameters",
7575
"sklearn kmeans parameters",
7676
"kmeans datasets"

‎configs/spmd/large_scale/full.json

Lines changed: 245 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,245 @@
1+
{
2+
"INCLUDE": ["../../common/sklearn.json", "large_scale.json"],
3+
"PARAMETERS_SETS": {
4+
"spmd basicstats parameters": {
5+
"algorithm": {
6+
"estimator": "BasicStatistics"
7+
}
8+
},
9+
"spmd incremental basicstats parameters": {
10+
"algorithm": {
11+
"estimator": "IncrementalBasicStatistics",
12+
"estimator_methods": {"training": "partial_fit"}
13+
}
14+
},
15+
"spmd covariance parameters": {
16+
"algorithm": {
17+
"estimator": "EmpiricalCovariance"
18+
}
19+
},
20+
"spmd incremental covariance parameters": {
21+
"algorithm": {
22+
"estimator": "IncrementalEmpiricalCovariance",
23+
"estimator_methods": {"training": "partial_fit"}
24+
}
25+
},
26+
"spmd pca parameters": {
27+
"algorithm": {
28+
"estimator": "PCA"
29+
}
30+
},
31+
"spmd incremental pca parameters": {
32+
"algorithm": {
33+
"estimator": "IncrementalPCA",
34+
"estimator_methods": {"training": "partial_fit"}
35+
}
36+
},
37+
"spmd dbscan parameters": {
38+
"algorithm": {
39+
"estimator": "DBSCAN"
40+
}
41+
},
42+
"spmd kmeans parameters": {
43+
"algorithm": {
44+
"estimator": "KMeans"
45+
},
46+
"estimator_params": {
47+
"algorithm": "lloyd"
48+
}
49+
},
50+
"spmd rfcls parameters": {
51+
"algorithm": {
52+
"estimator": "RandomForestClassifier"
53+
}
54+
},
55+
"spmd knncls parameters": {
56+
"algorithm": {
57+
"estimator": "KNeighborsClassifier"
58+
}
59+
},
60+
"spmd logreg parameters": {
61+
"algorithm": {
62+
"estimator": "LogisticRegression"
63+
},
64+
"estimator_params": {
65+
"solver": "newton-cg"
66+
}
67+
},
68+
"spmd rfreg parameters": {
69+
"algorithm": {
70+
"estimator": "RandomForestRegressor"
71+
}
72+
},
73+
"spmd knnreg parameters": {
74+
"algorithm": {
75+
"estimator": "KNeighborsRegressor"
76+
}
77+
},
78+
"spmd linreg parameters": {
79+
"algorithm": {
80+
"estimator": "LinearRegression"
81+
}
82+
},
83+
"spmd incremental linreg parameters": {
84+
"algorithm": {
85+
"estimator": "IncrementalLinearRegression",
86+
"estimator_methods": {"training": "partial_fit"}
87+
}
88+
},
89+
"synthetic stat data": {
90+
"data": [
91+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 1000, "n_features": 10, "centers": 1 } }
92+
]
93+
},
94+
"synthetic classification data": {
95+
"data": [
96+
{ "source": "make_classification", "split_kwargs": { "train_size": 1000, "test_size": 1000 }, "generation_kwargs": { "n_samples": 2000, "n_features": 10, "n_classes": 2 } }
97+
]
98+
},
99+
"synthetic regression data": {
100+
"data": [
101+
{ "source": "make_regression", "split_kwargs": { "train_size": 1000, "test_size": 1000 }, "generation_kwargs": { "n_samples": 2000, "n_features": 10 } }
102+
]
103+
},
104+
"universal": {
105+
"sklearnex_context": { "use_raw_input": [ true, false ] }
106+
}
107+
},
108+
"TEMPLATES": {
109+
"basicstats": {
110+
"SETS": [
111+
"sklearnex spmd implementation",
112+
"large scale default parameters",
113+
"synthetic stat data",
114+
"universal",
115+
"spmd basicstats parameters"
116+
]
117+
},
118+
"incremental basicstats": {
119+
"SETS": [
120+
"sklearnex spmd implementation",
121+
"large scale default parameters",
122+
"synthetic stat data",
123+
"universal",
124+
"spmd incremental basicstats parameters"
125+
]
126+
},
127+
"pca": {
128+
"SETS": [
129+
"sklearnex spmd implementation",
130+
"large scale default parameters",
131+
"synthetic stat data",
132+
"universal",
133+
"spmd pca parameters"
134+
]
135+
},
136+
"incremental pca": {
137+
"SETS": [
138+
"sklearnex spmd implementation",
139+
"large scale default parameters",
140+
"synthetic stat data",
141+
"universal",
142+
"spmd incremental pca parameters"
143+
]
144+
},
145+
"covariance": {
146+
"SETS": [
147+
"sklearnex spmd implementation",
148+
"large scale default parameters",
149+
"synthetic stat data",
150+
"universal",
151+
"spmd covariance parameters"
152+
]
153+
},
154+
"incremental covariance": {
155+
"SETS": [
156+
"sklearnex spmd implementation",
157+
"large scale default parameters",
158+
"synthetic stat data",
159+
"universal",
160+
"spmd incremental covariance parameters"
161+
]
162+
},
163+
"dbscan": {
164+
"SETS": [
165+
"sklearnex spmd implementation",
166+
"large scale default parameters",
167+
"synthetic stat data",
168+
"universal",
169+
"spmd dbscan parameters"
170+
]
171+
},
172+
"kmeans": {
173+
"SETS": [
174+
"sklearnex spmd implementation",
175+
"large scale default parameters",
176+
"synthetic stat data",
177+
"universal",
178+
"spmd kmeans parameters"
179+
]
180+
},
181+
"rfcls": {
182+
"SETS": [
183+
"sklearnex spmd implementation",
184+
"large scale default parameters",
185+
"synthetic classification data",
186+
"universal",
187+
"spmd rfcls parameters"
188+
]
189+
},
190+
"knncls": {
191+
"SETS": [
192+
"sklearnex spmd implementation",
193+
"large scale default parameters",
194+
"synthetic classification data",
195+
"universal",
196+
"spmd knncls parameters"
197+
]
198+
},
199+
"logreg": {
200+
"SETS": [
201+
"sklearnex spmd implementation",
202+
"large scale default parameters",
203+
"synthetic classification data",
204+
"universal",
205+
"spmd logreg parameters"
206+
]
207+
},
208+
"rfreg": {
209+
"SETS": [
210+
"sklearnex spmd implementation",
211+
"large scale default parameters",
212+
"synthetic regression data",
213+
"universal",
214+
"spmd rfreg parameters"
215+
]
216+
},
217+
"knnreg": {
218+
"SETS": [
219+
"sklearnex spmd implementation",
220+
"large scale default parameters",
221+
"synthetic regression data",
222+
"universal",
223+
"spmd knnreg parameters"
224+
]
225+
},
226+
"linreg": {
227+
"SETS": [
228+
"sklearnex spmd implementation",
229+
"large scale default parameters",
230+
"synthetic regression data",
231+
"universal",
232+
"spmd linreg parameters"
233+
]
234+
},
235+
"incremental linreg": {
236+
"SETS": [
237+
"sklearnex spmd implementation",
238+
"large scale default parameters",
239+
"synthetic regression data",
240+
"universal",
241+
"spmd incremental linreg parameters"
242+
]
243+
}
244+
}
245+
}

0 commit comments

Comments
 (0)