Skip to content

Commit 25dcf0c

Browse files
ethanglaserethanglaser
committed
re-add preview for lasso/enet
1 parent 5b6bbd3 commit 25dcf0c

4 files changed

Lines changed: 27 additions & 5 deletions

File tree

‎configs/common/sklearn.json‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,12 @@
1818
"data": { "format": "dpnp", "order": "C" }
1919
}
2020
],
21+
"sklearn-ex[preview,cpu] implementations": {
22+
"algorithm": [
23+
{ "library": "sklearn", "device": "cpu" },
24+
{ "library": "sklearnex.preview", "device": "cpu" }
25+
]
26+
},
2127
"sklearnex spmd implementation": {
2228
"algorithm": {
2329
"library": "sklearnex.spmd",

‎configs/regular/linear_model.json‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,10 +61,10 @@
6161
]
6262
},
6363
"sklearn lasso": {
64-
"SETS": ["sklearn-ex[cpu] implementations", "common lasso parameters", "regression datasets"]
64+
"SETS": ["sklearn-ex[preview,cpu] implementations", "common lasso parameters", "regression datasets"]
6565
},
6666
"sklearn elasticnet": {
67-
"SETS": ["sklearn-ex[cpu] implementations", "common elasticnet parameters", "regression datasets"]
67+
"SETS": ["sklearn-ex[preview,cpu] implementations", "common elasticnet parameters", "regression datasets"]
6868
},
6969
"cuml linear": {
7070
"SETS": ["cuml implementation", "common linear parameters", "cuml L2 parameters", "regression datasets"]

‎configs/testing/azure-pipelines-ci.json‎

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@
9191
},
9292
{
9393
"algorithm": {
94-
"estimator": ["LinearRegression", "Ridge", "Lasso", "ElasticNet"]
94+
"estimator": ["LinearRegression", "Ridge"]
9595
}
9696
},
9797
{
@@ -116,11 +116,27 @@
116116
}
117117
}
118118
}
119+
],
120+
"preview algorithms": [
121+
{
122+
"algorithm": {
123+
"estimator": ["Lasso", "ElasticNet"]
124+
}
125+
}
119126
]
120127
},
121128
"TEMPLATES": {
122129
"test": {
123130
"SETS": ["sklearn-ex[cpu] implementations", "common parameters", "data formats", "datasets", "algorithms"]
131+
},
132+
"test preview": {
133+
"SETS": [
134+
"sklearn-ex[preview,cpu] implementations",
135+
"common parameters",
136+
"data formats",
137+
"datasets",
138+
"preview algorithms"
139+
]
124140
}
125141
}
126142
}

‎configs/weekly/linear_model.json‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,10 +57,10 @@
5757
]
5858
},
5959
"sklearn lasso": {
60-
"SETS": ["sklearn-ex[cpu] implementations", "common lasso parameters", "regression datasets"]
60+
"SETS": ["sklearn-ex[preview,cpu] implementations", "common lasso parameters", "regression datasets"]
6161
},
6262
"sklearn elasticnet": {
63-
"SETS": ["sklearn-ex[cpu] implementations", "common elasticnet parameters", "regression datasets"]
63+
"SETS": ["sklearn-ex[preview,cpu] implementations", "common elasticnet parameters", "regression datasets"]
6464
},
6565
"cuml linear": {
6666
"SETS": ["cuml implementation", "common linear parameters", "cuml L2 parameters", "regression datasets"]

0 commit comments

Comments
 (0)