Skip to content

Commit 44cdb4f

Browse files
dmitrii-kriukovdmitrii-kriukov
authored
Configs for xpu: knn and svm (#106)
* knn_svm * epsilon dataset uses only train * fix epsilon * disable imb_drama * C-style format * Latin C * fix warning with higgs * resolve conflictsx2 * return covertype * too much cifar
1 parent 1c6b1d8 commit 44cdb4f

File tree

5 files changed

+590
-5
lines changed

5 files changed

+590
-5
lines changed

‎configs/xpu/knn_clsf.json

Lines changed: 162 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,162 @@
1+
{
2+
"common": {
3+
"lib": "sklearn",
4+
"algorithm": "knn_clsf",
5+
"data-format": "pandas",
6+
"data-order": "F",
7+
"dtype": ["float32", "float64"],
8+
"device": ["host", "cpu", "gpu", "none"]
9+
},
10+
"cases": [
11+
{
12+
"dataset": [
13+
{
14+
"source": "npy",
15+
"name": "epsilon_100K",
16+
"training":
17+
{
18+
"x": "data/epsilon_100K_x_train.npy",
19+
"y": "data/epsilon_100K_y_train.npy"
20+
},
21+
"testing":
22+
{
23+
"x": "data/epsilon_100K_x_train.npy",
24+
"y": "data/epsilon_100K_y_train.npy"
25+
}
26+
}
27+
],
28+
"n-neighbors": [2, 100]
29+
},
30+
{
31+
"dataset": [
32+
{
33+
"source": "npy",
34+
"name": "higgs_150K",
35+
"training":
36+
{
37+
"x": "data/higgs_150K_x_train.npy",
38+
"y": "data/higgs_150K_y_train.npy"
39+
},
40+
"testing":
41+
{
42+
"x": "data/higgs_150K_x_test.npy",
43+
"y": "data/higgs_150K_y_test.npy"
44+
}
45+
},
46+
{
47+
"source": "npy",
48+
"name": "hepmass_150K",
49+
"training":
50+
{
51+
"x": "data/hepmass_150K_x_train.npy",
52+
"y": "data/hepmass_150K_y_train.npy"
53+
},
54+
"testing":
55+
{
56+
"x": "data/hepmass_150K_x_test.npy",
57+
"y": "data/hepmass_150K_y_test.npy"
58+
}
59+
}
60+
],
61+
"n-neighbors": [5, 100]
62+
},
63+
{
64+
"dataset": [
65+
{
66+
"source": "npy",
67+
"name": "cifar_binary",
68+
"training":
69+
{
70+
"x": "data/cifar_binary_x_train.npy",
71+
"y": "data/cifar_binary_y_train.npy"
72+
},
73+
"testing":
74+
{
75+
"x": "data/cifar_binary_x_test.npy",
76+
"y": "data/cifar_binary_y_test.npy"
77+
}
78+
}
79+
],
80+
"n-neighbors": 7
81+
},
82+
{
83+
"dataset": [
84+
{
85+
"source": "npy",
86+
"name": "mnist",
87+
"training":
88+
{
89+
"x": "data/mnist_x_train.npy",
90+
"y": "data/mnist_y_train.npy"
91+
},
92+
"testing":
93+
{
94+
"x": "data/mnist_x_test.npy",
95+
"y": "data/mnist_y_test.npy"
96+
}
97+
}
98+
],
99+
"n-neighbors": 5
100+
},
101+
{
102+
"dataset": [
103+
{
104+
"source": "npy",
105+
"name": "epsilon_100K",
106+
"training":
107+
{
108+
"x": "data/epsilon_100K_x_train.npy",
109+
"y": "data/epsilon_100K_y_train.npy"
110+
},
111+
"testing":
112+
{
113+
"x": "data/epsilon_100K_x_train.npy",
114+
"y": "data/epsilon_100K_y_train.npy"
115+
}
116+
}
117+
],
118+
"task": "search",
119+
"n-neighbors": 2
120+
},
121+
{
122+
"dataset": [
123+
{
124+
"source": "npy",
125+
"name": "higgs_150K",
126+
"training":
127+
{
128+
"x": "data/higgs_150K_x_train.npy",
129+
"y": "data/higgs_150K_y_train.npy"
130+
},
131+
"testing":
132+
{
133+
"x": "data/higgs_150K_x_test.npy",
134+
"y": "data/higgs_150K_y_test.npy"
135+
}
136+
}
137+
],
138+
"task": "search",
139+
"n-neighbors": 5
140+
},
141+
{
142+
"dataset": [
143+
{
144+
"source": "npy",
145+
"name": "cifar_binary",
146+
"training":
147+
{
148+
"x": "data/cifar_binary_x_train.npy",
149+
"y": "data/cifar_binary_y_train.npy"
150+
},
151+
"testing":
152+
{
153+
"x": "data/cifar_binary_x_test.npy",
154+
"y": "data/cifar_binary_y_test.npy"
155+
}
156+
}
157+
],
158+
"task": "search",
159+
"n-neighbors": 7
160+
}
161+
]
162+
}

‎configs/xpu/knn_regr.json

Lines changed: 69 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,69 @@
1+
{
2+
"common": {
3+
"lib": "sklearn",
4+
"algorithm": "knn_regr",
5+
"data-format": "pandas",
6+
"data-order": "F",
7+
"dtype": ["float32", "float64"],
8+
"device": ["host", "cpu", "gpu", "none"]
9+
},
10+
"cases": [
11+
{
12+
"dataset": [
13+
{
14+
"source": "npy",
15+
"name": "year_prediction_msd",
16+
"training":
17+
{
18+
"x": "data/year_prediction_msd_x_train.npy",
19+
"y": "data/year_prediction_msd_y_train.npy"
20+
},
21+
"testing":
22+
{
23+
"x": "data/year_prediction_msd_x_test.npy",
24+
"y": "data/year_prediction_msd_y_test.npy"
25+
}
26+
}
27+
],
28+
"n-neighbors": 2
29+
},
30+
{
31+
"dataset": [
32+
{
33+
"source": "npy",
34+
"name": "higgs_150K",
35+
"training":
36+
{
37+
"x": "data/higgs_150K_x_train.npy",
38+
"y": "data/higgs_150K_y_train.npy"
39+
},
40+
"testing":
41+
{
42+
"x": "data/higgs_150K_x_test.npy",
43+
"y": "data/higgs_150K_y_test.npy"
44+
}
45+
}
46+
],
47+
"n-neighbors": 5
48+
},
49+
{
50+
"dataset": [
51+
{
52+
"source": "npy",
53+
"name": "cifar_binary",
54+
"training":
55+
{
56+
"x": "data/cifar_binary_x_train.npy",
57+
"y": "data/cifar_binary_y_train.npy"
58+
},
59+
"testing":
60+
{
61+
"x": "data/cifar_binary_x_test.npy",
62+
"y": "data/cifar_binary_y_test.npy"
63+
}
64+
}
65+
],
66+
"n-neighbors": 7
67+
}
68+
]
69+
}

‎configs/xpu/svm.json

Lines changed: 152 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,152 @@
1+
{
2+
"common": {
3+
"lib": "sklearn",
4+
"algorithm": "svm",
5+
"data-format": "pandas",
6+
"data-order": "C",
7+
"dtype": "float32",
8+
"device": ["host", "cpu", "gpu", "none"]
9+
},
10+
"cases": [
11+
{
12+
"dataset": [
13+
{
14+
"source": "npy",
15+
"name": "gisette",
16+
"training":
17+
{
18+
"x": "data/gisette_x_train.npy",
19+
"y": "data/gisette_y_train.npy"
20+
},
21+
"testing":
22+
{
23+
"x": "data/gisette_x_test.npy",
24+
"y": "data/gisette_y_test.npy"
25+
}
26+
}
27+
],
28+
"C": 1.5e-3,
29+
"kernel": "linear"
30+
},
31+
{
32+
"dataset": [
33+
{
34+
"source": "npy",
35+
"name": "higgs_150K",
36+
"training":
37+
{
38+
"x": "data/higgs_150K_x_train.npy",
39+
"y": "data/higgs_150K_y_train.npy"
40+
},
41+
"testing":
42+
{
43+
"x": "data/higgs_150K_x_test.npy",
44+
"y": "data/higgs_150K_y_test.npy"
45+
}
46+
}
47+
],
48+
"C": 1.0,
49+
"kernel": "linear"
50+
},
51+
{
52+
"dataset": [
53+
{
54+
"source": "npy",
55+
"name": "epsilon_80K",
56+
"training":
57+
{
58+
"x": "data/epsilon_80K_x_train.npy",
59+
"y": "data/epsilon_80K_y_train.npy"
60+
},
61+
"testing":
62+
{
63+
"x": "data/epsilon_80K_x_train.npy",
64+
"y": "data/epsilon_80K_y_train.npy"
65+
}
66+
}
67+
],
68+
"C": 1.0,
69+
"kernel": "linear"
70+
},
71+
{
72+
"dataset": [
73+
{
74+
"source": "npy",
75+
"name": "cifar_binary",
76+
"training":
77+
{
78+
"x": "data/cifar_binary_x_train.npy",
79+
"y": "data/cifar_binary_y_train.npy"
80+
},
81+
"testing":
82+
{
83+
"x": "data/cifar_binary_x_test.npy",
84+
"y": "data/cifar_binary_y_test.npy"
85+
}
86+
}
87+
],
88+
"C": 1.0e-7,
89+
"kernel": "linear"
90+
},
91+
{
92+
"dataset": [
93+
{
94+
"source": "npy",
95+
"name": "epsilon_16K",
96+
"training":
97+
{
98+
"x": "data/epsilon_16K_x_train.npy",
99+
"y": "data/epsilon_16K_y_train.npy"
100+
},
101+
"testing":
102+
{
103+
"x": "data/epsilon_16K_x_test.npy",
104+
"y": "data/epsilon_16K_y_test.npy"
105+
}
106+
}
107+
],
108+
"C": 9.0e2,
109+
"kernel": "rbf"
110+
},
111+
{
112+
"dataset": [
113+
{
114+
"source": "npy",
115+
"name": "covtype_binary",
116+
"training":
117+
{
118+
"x": "data/covtype_binary_x_train.npy",
119+
"y": "data/covtype_binary_y_train.npy"
120+
},
121+
"testing":
122+
{
123+
"x": "data/covtype_binary_x_test.npy",
124+
"y": "data/covtype_binary_y_test.npy"
125+
}
126+
}
127+
],
128+
"C": 1000.0,
129+
"kernel": "rbf"
130+
},
131+
{
132+
"dataset": [
133+
{
134+
"source": "npy",
135+
"name": "epsilon_80K",
136+
"training":
137+
{
138+
"x": "data/epsilon_80K_x_train.npy",
139+
"y": "data/epsilon_80K_y_train.npy"
140+
},
141+
"testing":
142+
{
143+
"x": "data/epsilon_80K_x_train.npy",
144+
"y": "data/epsilon_80K_y_train.npy"
145+
}
146+
}
147+
],
148+
"C": 1000.0,
149+
"kernel": "rbf"
150+
}
151+
]
152+
}

0 commit comments

Comments
 (0)