Skip to content

Commit 3988846

Browse files
dmitrii-kriukovdmitrii-kriukov
authored
Add size lables for xpu configs (#111)
* sizes in configs * test config * revert changing skl config * sizes for df_clsf * update help info * sizes for xpu configs
1 parent f2f84ad commit 3988846

File tree

9 files changed

+146
-10
lines changed

9 files changed

+146
-10
lines changed

‎configs/xpu/dbscan.json‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
}
2020
}
2121
],
22+
"workload-size": "small",
2223
"eps": 5,
2324
"min-samples": 3
2425
},
@@ -33,6 +34,7 @@
3334
}
3435
}
3536
],
37+
"workload-size": "medium",
3638
"eps": 1.7e3,
3739
"min-samples": 3
3840
},
@@ -47,6 +49,7 @@
4749
}
4850
}
4951
],
52+
"workload-size": "small",
5053
"eps": 1.0e3,
5154
"min-samples": 220
5255
}

‎configs/xpu/df_regr.json‎

Lines changed: 72 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@
2626
}
2727
}
2828
],
29-
"num-trees": [10, 100],
29+
"workload-size": "medium",
30+
"num-trees": 10,
3031
"max-depth": 5
3132
},
3233
{
@@ -46,7 +47,50 @@
4647
}
4748
}
4849
],
49-
"num-trees": [100, 20],
50+
"workload-size": "large",
51+
"num-trees": 10,
52+
"max-depth": 5
53+
},
54+
{
55+
"dataset": [
56+
{
57+
"source": "npy",
58+
"name": "year_prediction_msd",
59+
"training":
60+
{
61+
"x": "data/year_prediction_msd_x_train.npy",
62+
"y": "data/year_prediction_msd_y_train.npy"
63+
},
64+
"testing":
65+
{
66+
"x": "data/year_prediction_msd_x_test.npy",
67+
"y": "data/year_prediction_msd_y_test.npy"
68+
}
69+
}
70+
],
71+
"workload-size": "medium",
72+
"num-trees": 20,
73+
"max-depth": 8
74+
},
75+
{
76+
"dataset": [
77+
{
78+
"source": "npy",
79+
"name": "year_prediction_msd",
80+
"training":
81+
{
82+
"x": "data/year_prediction_msd_x_train.npy",
83+
"y": "data/year_prediction_msd_y_train.npy"
84+
},
85+
"testing":
86+
{
87+
"x": "data/year_prediction_msd_x_test.npy",
88+
"y": "data/year_prediction_msd_y_test.npy"
89+
}
90+
}
91+
],
92+
"workload-size": "large",
93+
"num-trees": 100,
5094
"max-depth": 8
5195
},
5296
{
@@ -66,6 +110,7 @@
66110
}
67111
}
68112
],
113+
"workload-size": "large",
69114
"num-trees": 20,
70115
"max-depth": 16
71116
},
@@ -86,7 +131,29 @@
86131
}
87132
}
88133
],
89-
"num-trees": [15, 20, 100],
134+
"workload-size": "medium",
135+
"num-trees": [15, 20],
136+
"max-depth": 8
137+
},
138+
{
139+
"dataset": [
140+
{
141+
"source": "npy",
142+
"name": "higgs1m",
143+
"training":
144+
{
145+
"x": "data/higgs1m_x_train.npy",
146+
"y": "data/higgs1m_y_train.npy"
147+
},
148+
"testing":
149+
{
150+
"x": "data/higgs1m_x_test.npy",
151+
"y": "data/higgs1m_y_test.npy"
152+
}
153+
}
154+
],
155+
"workload-size": "large",
156+
"num-trees": 100,
90157
"max-depth": 8
91158
},
92159
{
@@ -106,6 +173,7 @@
106173
}
107174
}
108175
],
176+
"workload-size": "large",
109177
"num-trees": 100,
110178
"max-depth": 8
111179
},
@@ -126,6 +194,7 @@
126194
}
127195
}
128196
],
197+
"workload-size": "large",
129198
"num-trees": 20,
130199
"max-depth": 16
131200
}

‎configs/xpu/kmeans.json‎

Lines changed: 40 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
}
2020
}
2121
],
22+
"workload-size": "medium",
2223
"n-clusters": 10,
2324
"maxiter": 100
2425
},
@@ -33,7 +34,23 @@
3334
}
3435
}
3536
],
36-
"n-clusters": [100, 250],
37+
"workload-size": "small",
38+
"n-clusters": 100,
39+
"maxiter": 10
40+
},
41+
{
42+
"dataset": [
43+
{
44+
"source": "npy",
45+
"name": "higgs_one_m_clustering",
46+
"training":
47+
{
48+
"x": "data/higgs_one_m_clustering.npy"
49+
}
50+
}
51+
],
52+
"workload-size": "medium",
53+
"n-clusters": 250,
3754
"maxiter": 10
3855
},
3956
{
@@ -47,6 +64,7 @@
4764
}
4865
}
4966
],
67+
"workload-size": "medium",
5068
"n-clusters": [512, 1024, 2048],
5169
"maxiter": 10
5270
},
@@ -61,7 +79,23 @@
6179
}
6280
}
6381
],
64-
"n-clusters": [100, 250],
82+
"workload-size": "small",
83+
"n-clusters": 100,
84+
"maxiter": 10
85+
},
86+
{
87+
"dataset": [
88+
{
89+
"source": "npy",
90+
"name": "hepmass_1M_cluster",
91+
"training":
92+
{
93+
"x": "data/hepmass_1M_cluster.npy"
94+
}
95+
}
96+
],
97+
"workload-size": "medium",
98+
"n-clusters": 250,
6599
"maxiter": 10
66100
},
67101
{
@@ -75,6 +109,7 @@
75109
}
76110
}
77111
],
112+
"workload-size": "medium",
78113
"n-clusters": 10,
79114
"maxiter": 100
80115
},
@@ -89,6 +124,7 @@
89124
}
90125
}
91126
],
127+
"workload-size": "medium",
92128
"n-clusters": 10,
93129
"maxiter": 100
94130
},
@@ -103,6 +139,7 @@
103139
}
104140
}
105141
],
142+
"workload-size": "medium",
106143
"n-clusters": [100 , 250],
107144
"maxiter": 10
108145
},
@@ -117,6 +154,7 @@
117154
}
118155
}
119156
],
157+
"workload-size": "medium",
120158
"n-clusters": [512, 1024, 2048],
121159
"maxiter": 10
122160
}

‎configs/xpu/knn_clsf.json‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
}
2626
}
2727
],
28+
"workload-size": "medium",
2829
"n-neighbors": [2, 100]
2930
},
3031
{
@@ -58,6 +59,7 @@
5859
}
5960
}
6061
],
62+
"workload-size": "medium",
6163
"n-neighbors": [5, 100]
6264
},
6365
{
@@ -77,6 +79,7 @@
7779
}
7880
}
7981
],
82+
"workload-size": "medium",
8083
"n-neighbors": 7
8184
},
8285
{
@@ -96,6 +99,7 @@
9699
}
97100
}
98101
],
102+
"workload-size": "medium",
99103
"n-neighbors": 5
100104
},
101105
{
@@ -115,6 +119,7 @@
115119
}
116120
}
117121
],
122+
"workload-size": "medium",
118123
"task": "search",
119124
"n-neighbors": 2
120125
},
@@ -135,6 +140,7 @@
135140
}
136141
}
137142
],
143+
"workload-size": "medium",
138144
"task": "search",
139145
"n-neighbors": 5
140146
},
@@ -155,6 +161,7 @@
155161
}
156162
}
157163
],
164+
"workload-size": "medium",
158165
"task": "search",
159166
"n-neighbors": 7
160167
}

‎configs/xpu/knn_regr.json‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
}
2626
}
2727
],
28+
"workload-size": "medium",
2829
"n-neighbors": 2
2930
},
3031
{
@@ -44,6 +45,7 @@
4445
}
4546
}
4647
],
48+
"workload-size": "medium",
4749
"n-neighbors": 5
4850
},
4951
{
@@ -63,6 +65,7 @@
6365
}
6466
}
6567
],
68+
"workload-size": "medium",
6669
"n-neighbors": 7
6770
}
6871
]

‎configs/xpu/linear.json‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,8 @@
2424
"y": "data/year_prediction_msd_y_test.npy"
2525
}
2626
}
27-
]
27+
],
28+
"workload-size": "small"
2829
},
2930
{
3031
"dataset": [
@@ -42,7 +43,8 @@
4243
"y": "data/higgs1m_y_test.npy"
4344
}
4445
}
45-
]
46+
],
47+
"workload-size": "small"
4648
}
4749
]
4850
}

‎configs/xpu/log_reg.json‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
}
2626
}
2727
],
28+
"workload-size": "medium",
2829
"maxiter": "20"
2930
},
3031
{
@@ -44,6 +45,7 @@
4445
}
4546
}
4647
],
48+
"workload-size": "medium",
4749
"maxiter": "10"
4850
},
4951
{
@@ -63,6 +65,7 @@
6365
}
6466
}
6567
],
68+
"workload-size": "small",
6669
"no-fit-intercept": "",
6770
"maxiter": "50"
6871
},
@@ -83,6 +86,7 @@
8386
}
8487
}
8588
],
89+
"workload-size": "medium",
8690
"maxiter": "500"
8791
}
8892
]

0 commit comments

Comments
 (0)