19
19
Supported Algorithms
20
20
####################
21
21
22
+ .. note ::
23
+ To verify that oneDAL is being used for these algorithms, you can enable verbose mode.
24
+ See :ref: `verbose mode documentation <verbose >` for details.
25
+
22
26
Applying |sklearnex | impacts the following |sklearn | estimators:
23
27
24
28
on CPU
@@ -48,6 +52,13 @@ Classification
48
52
- ``ccp_alpha `` != `0 `
49
53
- ``criterion `` != `'gini' `
50
54
- Multi-output and sparse data are not supported
55
+ * - `ExtraTreesClassifier `
56
+ - All parameters are supported except:
57
+
58
+ - ``warm_start `` = `True `
59
+ - ``ccp_alpha `` != `0 `
60
+ - ``criterion `` != `'gini' `
61
+ - Multi-output and sparse data are not supported
51
62
* - `KNeighborsClassifier `
52
63
-
53
64
- For ``algorithm `` == `'kd_tree' `:
@@ -58,12 +69,8 @@ Classification
58
69
all parameters except ``metric `` not in [`'euclidean' `, `'manhattan' `, `'minkowski' `, `'chebyshev' `, `'cosine' `]
59
70
- Multi-output and sparse data are not supported
60
71
* - `LogisticRegression `
61
- - All parameters are supported except:
62
-
63
- - ``solver `` not in [`'lbfgs' `, `'newton-cg' `]
64
- - ``class_weight `` != `None `
65
- - ``sample_weight `` != `None `
66
- - Only dense data is supported
72
+ - All parameters are supported
73
+ - No limitations
67
74
68
75
Regression
69
76
**********
@@ -89,6 +96,13 @@ Regression
89
96
- ``ccp_alpha `` != `0 `
90
97
- ``criterion `` != `'mse' `
91
98
- Multi-output and sparse data are not supported
99
+ * - `ExtraTreesRegressor `
100
+ - All parameters are supported except:
101
+
102
+ - ``warm_start `` = `True `
103
+ - ``ccp_alpha `` != `0 `
104
+ - ``criterion `` != `'mse' `
105
+ - Multi-output and sparse data are not supported
92
106
* - `KNeighborsRegressor `
93
107
- All parameters are supported except:
94
108
@@ -97,16 +111,17 @@ Regression
97
111
* - `LinearRegression `
98
112
- All parameters are supported except:
99
113
100
- - ``normalize `` != `False `
101
114
- ``sample_weight `` != `None `
115
+ - ``positive `` = `True `
102
116
- Only dense data is supported.
103
117
* - `Ridge `
104
118
- All parameters are supported except:
105
119
106
- - ``normalize `` != `False `
107
120
- ``solver `` != `'auto' `
108
121
- ``sample_weight `` != `None `
109
- - Only dense data is supported, `#observations ` should be >= `#features `.
122
+ - ``positive `` = `True `
123
+ - ``alpha `` must be scalar
124
+ - Only dense data is supported.
110
125
* - `ElasticNet `
111
126
- All parameters are supported except:
112
127
@@ -132,8 +147,10 @@ Clustering
132
147
* - `KMeans `
133
148
- All parameters are supported except:
134
149
135
- - ``precompute_distances ``
136
- - ``sample_weight `` != `None `
150
+ - ``algorithm `` != ``'lloyd' `` ('elkan' falls back to 'lloyd')
151
+ - ``n_clusters `` = ``1 ``
152
+ - ``sample_weight `` must be None, constant, or equal weights
153
+ - ``init `` = `'k-means++' ` falls back to CPU
137
154
- No limitations
138
155
* - `DBSCAN `
139
156
- All parameters are supported except:
@@ -156,16 +173,16 @@ Dimensionality Reduction
156
173
* - `PCA `
157
174
- All parameters are supported except:
158
175
159
- - ``svd_solver `` not in [`'full' `, `'covariance_eigh' `]
160
- - Sparse data is not supported
161
- * - `IncrementalPCA `
162
- - All parameters are supported
176
+ - ``svd_solver `` not in [`'full' `, `'covariance_eigh' `, `'onedal_svd' `]
177
+ - For |sklearn | < 1.5: `'full' ` solver is automatically mapped to `'covariance_eigh' `
163
178
- Sparse data is not supported
164
179
* - `TSNE `
165
180
- All parameters are supported except:
166
181
167
182
- ``metric `` != 'euclidean' or `'minkowski' ` with ``p `` != `2 `
168
183
184
+ - ``n_components `` can only be `2 `
185
+
169
186
Refer to :ref: `TSNE acceleration details <acceleration_tsne >` to learn more.
170
187
- Sparse data is not supported
171
188
@@ -204,17 +221,33 @@ Other Tasks
204
221
* - `EmpiricalCovariance `
205
222
- All parameters are supported
206
223
- Only dense data is supported
224
+ * - `BasicStatistics `
225
+ - All parameters are supported
226
+ - Supported data formats:
227
+
228
+ - Dense data
229
+ - CSR sparse matrices
230
+ - Sample weights **not ** supported for CSR data format
207
231
* - `train_test_split `
208
232
- All parameters are supported
209
- - Only dense data is supported
233
+ - Supported data formats:
234
+
235
+ - Only dense data is supported
236
+ - Only integer and 32/64-bits floating point types are supported
237
+ - Data with more than 3 dimensions is not supported
238
+ - Only ``np.ndarray `` inputs are supported.
210
239
* - `assert_all_finite `
211
240
- All parameters are supported
212
241
- Only dense data is supported
213
242
* - `pairwise_distance `
214
243
- All parameters are supported except:
215
244
216
245
- ``metric `` not in [`'cosine' `, `'correlation' `]
217
- - Only dense data is supported
246
+ - Supported data formats:
247
+
248
+ - Only dense data is supported
249
+ - ``Y `` must be `None `
250
+ - Input dtype must be `np.float64 `
218
251
* - `roc_auc_score `
219
252
- All parameters are supported except:
220
253
@@ -255,6 +288,15 @@ Classification
255
288
- ``oob_score `` = `True `
256
289
- ``sample_weight `` != `None `
257
290
- Multi-output and sparse data are not supported
291
+ * - `ExtraTreesClassifier `
292
+ - All parameters are supported except:
293
+
294
+ - ``warm_start `` = `True `
295
+ - ``ccp_alpha `` != `0 `
296
+ - ``criterion `` != `'gini' `
297
+ - ``oob_score `` = `True `
298
+ - ``sample_weight `` != `None `
299
+ - Multi-output and sparse data are not supported
258
300
* - `KNeighborsClassifier `
259
301
- All parameters are supported except:
260
302
@@ -269,7 +311,13 @@ Classification
269
311
- ``class_weight `` != `None `
270
312
- ``sample_weight `` != `None `
271
313
- ``penalty `` != `'l2' `
272
- - Only dense data is supported
314
+ - ``dual `` = `True `
315
+ - ``intercept_scaling `` != `1 `
316
+ - ``multi_class `` != `'multinomial' `
317
+ - ``warm_start `` = `True `
318
+ - ``l1_ratio `` != `None `
319
+ - Only binary classification is supported
320
+ - No limitations
273
321
274
322
Regression
275
323
**********
@@ -291,6 +339,15 @@ Regression
291
339
- ``oob_score `` = `True `
292
340
- ``sample_weight `` != `None `
293
341
- Multi-output and sparse data are not supported
342
+ * - `ExtraTreesRegressor `
343
+ - All parameters are supported except:
344
+
345
+ - ``warm_start `` = `True `
346
+ - ``ccp_alpha `` != `0 `
347
+ - ``criterion `` != `'mse' `
348
+ - ``oob_score `` = `True `
349
+ - ``sample_weight `` != `None `
350
+ - Multi-output and sparse data are not supported
294
351
* - `KNeighborsRegressor `
295
352
- All parameters are supported except:
296
353
@@ -301,8 +358,8 @@ Regression
301
358
* - `LinearRegression `
302
359
- All parameters are supported except:
303
360
304
- - ``normalize `` != `False `
305
361
- ``sample_weight `` != `None `
362
+ - ``positive `` = `True `
306
363
- Only dense data is supported.
307
364
308
365
Clustering
@@ -319,10 +376,11 @@ Clustering
319
376
* - `KMeans `
320
377
- All parameters are supported except:
321
378
322
- - ``precompute_distances ``
323
- - ``sample_weight `` != `None `
324
- - ``Init `` = `'k-means++' ` fallbacks to CPU.
325
- - Sparse data is not supported
379
+ - ``algorithm `` != ``'lloyd' `` ('elkan' falls back to 'lloyd')
380
+ - ``n_clusters `` = ``1 ``
381
+ - ``sample_weight `` must be None, constant, or equal weights
382
+ - ``init `` = `'k-means++' ` falls back to CPU
383
+ - No limitations
326
384
* - `DBSCAN `
327
385
- All parameters are supported except:
328
386
@@ -344,7 +402,8 @@ Dimensionality Reduction
344
402
* - `PCA `
345
403
- All parameters are supported except:
346
404
347
- - ``svd_solver `` not in [`'full' `, `'covariance_eigh' `]
405
+ - ``svd_solver `` not in [`'full' `, `'covariance_eigh' `, `'onedal_svd' `]
406
+ - For |sklearn | < 1.5: `'full' ` solver is automatically mapped to `'covariance_eigh' `
348
407
- Sparse data is not supported
349
408
350
409
Nearest Neighbors
@@ -380,6 +439,13 @@ Other Tasks
380
439
* - `EmpiricalCovariance `
381
440
- All parameters are supported
382
441
- Only dense data is supported
442
+ * - `BasicStatistics `
443
+ - All parameters are supported
444
+ - Supported data formats:
445
+
446
+ - Dense data
447
+ - CSR sparse matrices
448
+ - Sample weights **not ** supported for CSR data format
383
449
384
450
.. _spmd-support :
385
451
@@ -408,6 +474,15 @@ Classification
408
474
- ``oob_score `` = `True `
409
475
- ``sample_weight `` != `None `
410
476
- Multi-output and sparse data are not supported
477
+ * - `ExtraTreesClassifier `
478
+ - All parameters are supported except:
479
+
480
+ - ``warm_start `` = `True `
481
+ - ``ccp_alpha `` != `0 `
482
+ - ``criterion `` != `'gini' `
483
+ - ``oob_score `` = `True `
484
+ - ``sample_weight `` != `None `
485
+ - Multi-output and sparse data are not supported
411
486
* - `KNeighborsClassifier `
412
487
- All parameters are supported except:
413
488
@@ -423,7 +498,13 @@ Classification
423
498
- ``class_weight `` != `None `
424
499
- ``sample_weight `` != `None `
425
500
- ``penalty `` != `'l2' `
426
- - Only dense data is supported
501
+ - ``dual `` = `True `
502
+ - ``intercept_scaling `` != `1 `
503
+ - ``multi_class `` != `'multinomial' `
504
+ - ``warm_start `` = `True `
505
+ - ``l1_ratio `` != `None `
506
+ - Only binary classification is supported
507
+ - No limitations
427
508
428
509
Regression
429
510
**********
@@ -445,6 +526,15 @@ Regression
445
526
- ``oob_score `` = `True `
446
527
- ``sample_weight `` != `None `
447
528
- Multi-output and sparse data are not supported
529
+ * - `ExtraTreesRegressor `
530
+ - All parameters are supported except:
531
+
532
+ - ``warm_start `` = `True `
533
+ - ``ccp_alpha `` != `0 `
534
+ - ``criterion `` != `'mse' `
535
+ - ``oob_score `` = `True `
536
+ - ``sample_weight `` != `None `
537
+ - Multi-output and sparse data are not supported
448
538
* - `KNeighborsRegressor `
449
539
- All parameters are supported except:
450
540
@@ -455,8 +545,8 @@ Regression
455
545
* - `LinearRegression `
456
546
- All parameters are supported except:
457
547
458
- - ``normalize `` != `False `
459
548
- ``sample_weight `` != `None `
549
+ - ``positive `` = `True `
460
550
- Only dense data is supported.
461
551
462
552
Clustering
@@ -473,10 +563,11 @@ Clustering
473
563
* - `KMeans `
474
564
- All parameters are supported except:
475
565
476
- - ``precompute_distances ``
477
- - ``sample_weight `` != `None `
478
- - ``Init `` = `'k-means++' ` fallbacks to CPU.
479
- - Sparse data is not supported
566
+ - ``algorithm `` != ``'lloyd' `` ('elkan' falls back to 'lloyd')
567
+ - ``n_clusters `` = ``1 ``
568
+ - ``sample_weight `` must be None, constant, or equal weights
569
+ - ``init `` = `'k-means++' ` falls back to CPU
570
+ - No limitations
480
571
* - `DBSCAN `
481
572
- All parameters are supported except:
482
573
@@ -498,8 +589,8 @@ Dimensionality Reduction
498
589
* - `PCA `
499
590
- All parameters are supported except:
500
591
501
- - ``svd_solver `` not in [`'full' `, `'covariance_eigh' `]
502
- - `` fit `` is the only method supported
592
+ - ``svd_solver `` not in [`'full' `, `'covariance_eigh' `, ` 'onedal_svd' ` ]
593
+ - For | sklearn | < 1.5: ` 'full' ` solver is automatically mapped to ` 'covariance_eigh' `
503
594
- Sparse data is not supported
504
595
505
596
Nearest Neighbors
@@ -535,6 +626,13 @@ Other Tasks
535
626
* - `EmpiricalCovariance `
536
627
- All parameters are supported
537
628
- Only dense data is supported
629
+ * - `BasicStatistics `
630
+ - All parameters are supported
631
+ - Supported data formats:
632
+
633
+ - Dense data
634
+ - CSR sparse matrices
635
+ - Sample weights **not ** supported for CSR data format
538
636
539
637
Scikit-learn Tests
540
638
------------------
0 commit comments