Skip to content

Commit 6272ea5

Browse files
dmitrii-kriukovdmitrii-kriukov
committed
knn_svm
1 parent 2c1fec5 commit 6272ea5

File tree

5 files changed

+722
-4
lines changed

5 files changed

+722
-4
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_test.npy",
24+
"y": "data/epsilon_100K_y_test.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",
68+
"training":
69+
{
70+
"x": "data/cifar_x_train.npy",
71+
"y": "data/cifar_y_train.npy"
72+
},
73+
"testing":
74+
{
75+
"x": "data/cifar_x_test.npy",
76+
"y": "data/cifar_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_test.npy",
114+
"y": "data/epsilon_100K_y_test.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",
146+
"training":
147+
{
148+
"x": "data/cifar_x_train.npy",
149+
"y": "data/cifar_y_train.npy"
150+
},
151+
"testing":
152+
{
153+
"x": "data/cifar_x_test.npy",
154+
"y": "data/cifar_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",
54+
"training":
55+
{
56+
"x": "data/cifar_x_train.npy",
57+
"y": "data/cifar_y_train.npy"
58+
},
59+
"testing":
60+
{
61+
"x": "data/cifar_x_test.npy",
62+
"y": "data/cifar_y_test.npy"
63+
}
64+
}
65+
],
66+
"n-neighbors": 7
67+
}
68+
]
69+
}

0 commit comments

Comments
 (0)