Skip to content

Commit fd6578e

Browse files
AlexsandrussAlexsandruss
committed
Add configs
1 parent bdcedf1 commit fd6578e

File tree

4 files changed

+589
-4
lines changed

4 files changed

+589
-4
lines changed

‎configs/cuml_config.json

Lines changed: 290 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,290 @@
1+
{
2+
"omp_env": ["OMP_NUM_THREADS"],
3+
"common": {
4+
"lib": ["cuml"],
5+
"data-format": ["cudf"],
6+
"data-order": ["F"],
7+
"dtype": ["float64"]
8+
},
9+
"cases": [
10+
{
11+
"algorithm": "kmeans",
12+
"dataset": [
13+
{
14+
"source": "synthetic",
15+
"type": "blobs",
16+
"n_clusters": 1000,
17+
"n_features": 20,
18+
"training": {
19+
"n_samples": 1000000
20+
}
21+
}
22+
],
23+
"time-method": ["box_filter"],
24+
"n-clusters": [1000],
25+
"maxiter": [50],
26+
"tol": [0.0]
27+
},
28+
{
29+
"algorithm": "kmeans",
30+
"dataset": [
31+
{
32+
"source": "synthetic",
33+
"type": "blobs",
34+
"n_clusters": 5,
35+
"n_features": 50,
36+
"training": {
37+
"n_samples": 10000000
38+
}
39+
}
40+
],
41+
"time-method": ["box_filter"],
42+
"n-clusters": [5],
43+
"maxiter": [50],
44+
"init": ["k-means++"],
45+
"tol": [0.0]
46+
},
47+
{
48+
"algorithm": "kmeans",
49+
"dataset": [
50+
{
51+
"source": "synthetic",
52+
"type": "blobs",
53+
"n_clusters": 20,
54+
"n_features": 50,
55+
"training": {
56+
"n_samples": 3000000
57+
}
58+
}
59+
],
60+
"time-method": ["box_filter"],
61+
"n-clusters": [20],
62+
"maxiter": [50],
63+
"tol": [0.0]
64+
},
65+
{
66+
"algorithm": "pca",
67+
"dataset": [
68+
{
69+
"source": "synthetic",
70+
"type": "classification",
71+
"n_classes": 2,
72+
"n_features": 100,
73+
"training": {
74+
"n_samples": 1000000
75+
},
76+
"testing": {
77+
"n_samples": 100000
78+
}
79+
},
80+
{
81+
"source": "synthetic",
82+
"type": "classification",
83+
"n_classes": 2,
84+
"n_features": 2000,
85+
"training": {
86+
"n_samples": 10000
87+
}
88+
},
89+
{
90+
"source": "synthetic",
91+
"type": "classification",
92+
"n_classes": 2,
93+
"n_features": 1000,
94+
"training": {
95+
"n_samples": 30000
96+
}
97+
}
98+
],
99+
"svd-solver": ["full"],
100+
"n-components": [10]
101+
},
102+
{
103+
"algorithm": "ridge",
104+
"dataset": [
105+
{
106+
"source": "synthetic",
107+
"type": "regression",
108+
"n_features": 20,
109+
"training": {
110+
"n_samples": 10000000
111+
}
112+
},
113+
{
114+
"source": "synthetic",
115+
"type": "regression",
116+
"n_features": 100,
117+
"training": {
118+
"n_samples": 2000000
119+
}
120+
}
121+
],
122+
"alpha": [5]
123+
},
124+
{
125+
"algorithm": "linear",
126+
"dataset": [
127+
{
128+
"source": "synthetic",
129+
"type": "regression",
130+
"n_features": 20,
131+
"training": {
132+
"n_samples": 10000000
133+
}
134+
},
135+
{
136+
"source": "synthetic",
137+
"type": "regression",
138+
"n_features": 100,
139+
"training": {
140+
"n_samples": 2000000
141+
}
142+
}
143+
]
144+
},
145+
{
146+
"algorithm": "log_reg",
147+
"dataset": [
148+
{
149+
"source": "synthetic",
150+
"type": "classification",
151+
"n_classes": 2,
152+
"n_features": 20,
153+
"training": {
154+
"n_samples": 10000000
155+
}
156+
},
157+
{
158+
"source": "synthetic",
159+
"type": "classification",
160+
"n_classes": 2,
161+
"n_features": 100,
162+
"training": {
163+
"n_samples": 2000000
164+
}
165+
},
166+
{
167+
"source": "synthetic",
168+
"type": "classification",
169+
"n_classes": 5,
170+
"n_features": 20,
171+
"training": {
172+
"n_samples": 10000000
173+
}
174+
},
175+
{
176+
"source": "synthetic",
177+
"type": "classification",
178+
"n_classes": 5,
179+
"n_features": 100,
180+
"training": {
181+
"n_samples": 2000000
182+
}
183+
}
184+
],
185+
"maxiter": [100],
186+
"tol": [0]
187+
},
188+
{
189+
"algorithm": "dbscan",
190+
"dataset": [
191+
{
192+
"source": "synthetic",
193+
"type": "blobs",
194+
"n_clusters": 50,
195+
"n_features": 3,
196+
"training": {
197+
"n_samples": 500000
198+
}
199+
},
200+
{
201+
"source": "synthetic",
202+
"type": "blobs",
203+
"n_clusters": 50,
204+
"n_features": 10,
205+
"training": {
206+
"n_samples": 500000
207+
}
208+
},
209+
{
210+
"source": "synthetic",
211+
"type": "blobs",
212+
"n_clusters": 100,
213+
"n_features": 50,
214+
"training": {
215+
"n_samples": 500000
216+
}
217+
}
218+
]
219+
},
220+
{
221+
"algorithm": "knn_clsf",
222+
"dtype": ["float32"],
223+
"dataset": [
224+
{
225+
"source": "synthetic",
226+
"type": "classification",
227+
"n_classes": 2,
228+
"n_features": 3,
229+
"training": {
230+
"n_samples": 100000
231+
}
232+
},
233+
{
234+
"source": "synthetic",
235+
"type": "classification",
236+
"n_classes": 2,
237+
"n_features": 10,
238+
"training": {
239+
"n_samples": 100000
240+
}
241+
},
242+
{
243+
"source": "synthetic",
244+
"type": "classification",
245+
"n_classes": 2,
246+
"n_features": 50,
247+
"training": {
248+
"n_samples": 20000
249+
}
250+
}
251+
],
252+
"weights": ["uniform"],
253+
"method": ["kd_tree", "brute"]
254+
},
255+
{
256+
"algorithm": "train_test_split",
257+
"dataset": [
258+
{
259+
"source": "synthetic",
260+
"type": "classification",
261+
"n_classes": 2,
262+
"n_features": 20,
263+
"training": {
264+
"n_samples": 5000000
265+
}
266+
},
267+
{
268+
"source": "synthetic",
269+
"type": "classification",
270+
"n_classes": 2,
271+
"n_features": 100,
272+
"training": {
273+
"n_samples": 1000000
274+
}
275+
},
276+
{
277+
"source": "synthetic",
278+
"type": "classification",
279+
"n_classes": 2,
280+
"n_features": 10000,
281+
"training": {
282+
"n_samples": 10000
283+
}
284+
}
285+
],
286+
"train-size": [0.75],
287+
"test-size": [0.25]
288+
}
289+
]
290+
}

0 commit comments

Comments
 (0)