Skip to content

Commit f3be737

Browse files
md-shafiul-alamolegkkruglov
authored andcommitted
just gpu for regular
1 parent b1f2c15 commit f3be737

File tree

8 files changed

+14
-47
lines changed

8 files changed

+14
-47
lines changed

configs/common/sklearn.json

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,11 @@
1212
{ "library": "sklearnex", "device": ["cpu", "gpu"] }
1313
]
1414
},
15+
"sklearn-ex[gpu] implementations": {
16+
"algorithm": [
17+
{ "library": "sklearnex", "device": ["gpu"] }
18+
]
19+
},
1520
"sklearn-ex[preview] implementations": {
1621
"algorithm": [
1722
{ "library": "sklearn", "device": "cpu" },

configs/regular/dbscan.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@
5858
"TEMPLATES": {
5959
"sklearn dbscan": {
6060
"SETS": [
61-
"sklearn-ex[cpu,gpu] implementations",
61+
"sklearn-ex[gpu] implementations",
6262
"common dbscan parameters",
6363
"sklearn dbscan parameters",
6464
"dbscan datasets"

configs/regular/ensemble.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,15 +90,15 @@
9090
"TEMPLATES": {
9191
"sklearn ensemble classification": {
9292
"SETS": [
93-
"sklearn-ex[cpu,gpu] implementations",
93+
"sklearn-ex[gpu] implementations",
9494
"common ensemble params",
9595
"sklearn ensemble classifier params",
9696
"ensemble classification data"
9797
]
9898
},
9999
"sklearn ensemble regression": {
100100
"SETS": [
101-
"sklearn-ex[cpu,gpu] implementations",
101+
"sklearn-ex[gpu] implementations",
102102
"common ensemble params",
103103
"sklearn ensemble regressor params",
104104
"ensemble regression data"

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[gpu] implementations",
7474
"common kmeans parameters",
7575
"sklearn kmeans parameters",
7676
"kmeans datasets"

configs/regular/knn.json

Lines changed: 2 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -74,36 +74,20 @@
7474
"TEMPLATES": {
7575
"sklearn brute knn clsf": {
7676
"SETS": [
77-
"sklearn-ex[cpu,gpu] implementations",
77+
"sklearn-ex[gpu] implementations",
7878
"common knn parameters",
7979
"sklearn knn parameters",
8080
"brute knn algorithm - classification data"
8181
]
8282
},
83-
"sklearn kd_tree knn clsf": {
84-
"SETS": [
85-
"sklearn-ex[cpu] implementations",
86-
"common knn parameters",
87-
"sklearn knn parameters",
88-
"kd_tree knn algorithm - classification data"
89-
]
90-
},
9183
"sklearn brute knn regr": {
9284
"SETS": [
93-
"sklearn-ex[cpu,gpu] implementations",
85+
"sklearn-ex[gpu] implementations",
9486
"common knn parameters",
9587
"sklearn knn parameters",
9688
"brute knn algorithm - regression data"
9789
]
9890
},
99-
"sklearn kd_tree knn regr": {
100-
"SETS": [
101-
"sklearn-ex[cpu] implementations",
102-
"common knn parameters",
103-
"sklearn knn parameters",
104-
"kd_tree knn algorithm - regression data"
105-
]
106-
},
10791
"cuml brute knn clsf": {
10892
"SETS": [
10993
"cuml implementation",

configs/regular/linear_model.json

Lines changed: 1 addition & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -85,34 +85,12 @@
8585
"TEMPLATES": {
8686
"sklearn linear": {
8787
"SETS": [
88-
"sklearn-ex[cpu,gpu] implementations",
88+
"sklearn-ex[gpu] implementations",
8989
"common linear parameters",
9090
"sklearn linear parameters",
9191
"regression datasets"
9292
]
9393
},
94-
"sklearn ridge": {
95-
"SETS": [
96-
"sklearn-ex[cpu] implementations",
97-
"common ridge parameters",
98-
"sklearn ridge parameters",
99-
"regression datasets"
100-
]
101-
},
102-
"sklearn lasso": {
103-
"SETS": [
104-
"sklearn-ex[cpu] implementations",
105-
"common lasso parameters",
106-
"regression datasets"
107-
]
108-
},
109-
"sklearn elasticnet": {
110-
"SETS": [
111-
"sklearn-ex[cpu] implementations",
112-
"common elasticnet parameters",
113-
"regression datasets"
114-
]
115-
},
11694
"cuml linear": {
11795
"SETS": [
11896
"cuml implementation",

configs/regular/logreg.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454
"TEMPLATES": {
5555
"sklearn logreg": {
5656
"SETS": [
57-
"sklearn-ex[cpu,gpu] implementations",
57+
"sklearn-ex[gpu] implementations",
5858
"common logreg parameters",
5959
"sklearn logreg parameters",
6060
"logreg datasets"

configs/regular/pca.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646
"TEMPLATES": {
4747
"sklearn pca": {
4848
"SETS": [
49-
"sklearn-ex[cpu,gpu] implementations",
49+
"sklearn-ex[gpu] implementations",
5050
"pca parameters",
5151
"pca datasets"
5252
]

0 commit comments

Comments
 (0)