scikit-learn-intelex 2024.4.0__py312-none-win_amd64.whl → 2025.10.0__py312-none-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (299) hide show
  1. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/__init__.py +73 -0
  2. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/__main__.py +58 -0
  3. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/_daal4py.cp312-win_amd64.pyd +0 -0
  4. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mb/__init__.py +94 -0
  5. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mb/gbt_convertors.py +1199 -0
  6. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mb/logistic_regression_builders.py +211 -0
  7. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mb/tree_based_builders.py +425 -0
  8. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mpi_transceiver.cp312-win_amd64.pyd +0 -0
  9. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/__init__.py +40 -0
  10. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/_n_jobs_support.py +252 -0
  11. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/_utils.py +245 -0
  12. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/__init__.py +20 -0
  13. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/dbscan.py +165 -0
  14. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/k_means.py +565 -0
  15. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/tests/test_dbscan.py +109 -0
  16. {scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn}/decomposition/__init__.py +2 -2
  17. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/decomposition/_pca.py +528 -0
  18. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/AdaBoostClassifier.py +192 -0
  19. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/GBTDAAL.py +333 -0
  20. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/__init__.py +27 -0
  21. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/_forest.py +1285 -0
  22. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/tests/test_decision_forest.py +206 -0
  23. {scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn}/linear_model/__init__.py +29 -28
  24. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_coordinate_descent.py +826 -0
  25. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_linear.py +269 -0
  26. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_ridge.py +290 -0
  27. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/basic_statistics/basic_statistics.py → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/coordinate_descent.py +2 -2
  28. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/linear.py +17 -0
  29. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/logistic_loss.py +195 -0
  30. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/logistic_path.py +561 -0
  31. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/ridge.py +17 -0
  32. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_enet.py +157 -0
  33. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_linear.py +198 -0
  34. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_ridge.py +69 -0
  35. {scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/preview/cluster → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/manifold}/__init__.py +3 -3
  36. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/manifold/_t_sne.py +432 -0
  37. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/metrics/__init__.py +20 -0
  38. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/metrics/_pairwise.py +259 -0
  39. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/metrics/_ranking.py +210 -0
  40. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/ridge.py → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/model_selection/__init__.py +4 -2
  41. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/model_selection/_split.py +309 -0
  42. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/model_selection/tests/test_split.py +56 -0
  43. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/__init__.py +0 -0
  44. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/dispatcher.py +232 -0
  45. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/_models_info.py +164 -0
  46. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/test_monkeypatch.py +71 -0
  47. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/test_patching.py +90 -0
  48. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/utils/_launch_algorithms.py +117 -0
  49. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/__init__.py +21 -0
  50. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_base.py +493 -0
  51. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_classification.py +136 -0
  52. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_regression.py +72 -0
  53. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_unsupervised.py +55 -0
  54. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/tests/test_kneighbors.py +113 -0
  55. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/svm/__init__.py +19 -0
  56. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/svm/svm.py +736 -0
  57. {scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd/covariance → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/utils}/__init__.py +5 -3
  58. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/utils/base.py +80 -0
  59. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/utils/tests/test_utils.py +51 -0
  60. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/utils/validation.py +772 -0
  61. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/__init__.py +151 -0
  62. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/_config.py +80 -0
  63. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/_device_offload.py +199 -0
  64. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/_onedal_py_dpc.cp312-win_amd64.pyd +0 -0
  65. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/_onedal_py_host.cp312-win_amd64.pyd +0 -0
  66. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/basic_statistics.py +151 -0
  67. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/incremental_basic_statistics.py +165 -0
  68. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/tests/test_basic_statistics.py +241 -0
  69. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/tests/test_incremental_basic_statistics.py +279 -0
  70. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/tests/utils.py +50 -0
  71. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/__init__.py +27 -0
  72. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/dbscan.py +80 -0
  73. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/kmeans.py +582 -0
  74. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/kmeans_init.py +145 -0
  75. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/tests/test_dbscan.py +125 -0
  76. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/tests/test_kmeans.py +88 -0
  77. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/tests/test_kmeans_init.py +93 -0
  78. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/_backend.py +258 -0
  79. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/_estimator_checks.py +47 -0
  80. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/_mixin.py +62 -0
  81. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/hyperparameters.py +148 -0
  82. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/tests/test_sycl.py +148 -0
  83. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/__init__.py +20 -0
  84. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/covariance.py +121 -0
  85. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/incremental_covariance.py +151 -0
  86. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/tests/test_covariance.py +50 -0
  87. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/tests/test_incremental_covariance.py +190 -0
  88. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/__init__.py +29 -0
  89. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/_data_conversion.py +158 -0
  90. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/_dlpack.py +64 -0
  91. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/_sycl_usm.py +63 -0
  92. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/tests/common.py +131 -0
  93. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/tests/test_data.py +686 -0
  94. {scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/decomposition}/__init__.py +3 -2
  95. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/decomposition/incremental_pca.py +218 -0
  96. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/decomposition/pca.py +124 -0
  97. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/decomposition/tests/test_incremental_pca.py +291 -0
  98. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/dummy/__init__.py +19 -0
  99. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/dummy/dummy.py +137 -0
  100. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/ensemble/__init__.py +29 -0
  101. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/ensemble/forest.py +781 -0
  102. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/ensemble/tests/test_random_forest.py +97 -0
  103. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/__init__.py +27 -0
  104. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/incremental_linear_model.py +201 -0
  105. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/linear_model.py +230 -0
  106. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/logistic_regression.py +293 -0
  107. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_incremental_linear_regression.py +213 -0
  108. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_incremental_ridge_regression.py +171 -0
  109. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_linear_regression.py +252 -0
  110. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_logistic_regression.py +95 -0
  111. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_ridge.py +95 -0
  112. {scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal}/neighbors/__init__.py +19 -19
  113. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/neighbors/neighbors.py +690 -0
  114. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/neighbors/tests/test_knn_classification.py +49 -0
  115. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/primitives/__init__.py +27 -0
  116. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/primitives/get_tree.py +27 -0
  117. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/primitives/kernel_functions.py +202 -0
  118. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/primitives/tests/test_kernel_functions.py +159 -0
  119. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/__init__.py +19 -0
  120. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/svm.py +592 -0
  121. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_csr_svm.py +352 -0
  122. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_nusvc.py +204 -0
  123. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_nusvr.py +210 -0
  124. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_svc.py +168 -0
  125. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_svr.py +243 -0
  126. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/tests/test_common.py +71 -0
  127. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/tests/utils/_dataframes_support.py +179 -0
  128. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/tests/utils/_device_selection.py +94 -0
  129. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/_array_api.py +98 -0
  130. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/_sycl_queue_manager.py +213 -0
  131. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/_third_party.py +220 -0
  132. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/tests/test_validation.py +142 -0
  133. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/validation.py +503 -0
  134. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/__init__.py +7 -3
  135. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/__main__.py +2 -2
  136. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/_config.py +163 -0
  137. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/_device_offload.py +205 -0
  138. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/_utils.py +219 -0
  139. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/base.py +109 -0
  140. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/basic_statistics/__init__.py +20 -0
  141. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/basic_statistics/basic_statistics.py +241 -0
  142. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/basic_statistics/incremental_basic_statistics.py +128 -78
  143. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/basic_statistics/tests/test_basic_statistics.py +405 -0
  144. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/basic_statistics/tests/test_incremental_basic_statistics.py +101 -32
  145. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/cluster/__init__.py +1 -1
  146. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/cluster/dbscan.py +38 -29
  147. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/cluster/k_means.py +399 -0
  148. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/cluster/tests/test_dbscan.py +8 -6
  149. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/cluster/tests/test_kmeans.py +157 -0
  150. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/conftest.py +20 -1
  151. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/covariance/incremental_covariance.py +440 -0
  152. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/covariance/tests/test_incremental_covariance.py +307 -0
  153. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/decomposition/pca.py +558 -0
  154. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/decomposition/tests/test_pca.py +164 -0
  155. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/dispatcher.py +199 -21
  156. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/doc/third-party-programs.txt +207 -2
  157. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/coordinate_descent.py → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/dummy/__init__.py +19 -17
  158. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/dummy/_dummy.py +615 -0
  159. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/dummy/tests/test_dummy.py +62 -0
  160. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/ensemble/_forest.py +288 -440
  161. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/ensemble/tests/test_forest.py +196 -0
  162. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/glob/__main__.py +1 -1
  163. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/glob/dispatcher.py +17 -3
  164. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/__init__.py +32 -0
  165. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/coordinate_descent.py +44 -0
  166. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/incremental_linear.py +427 -0
  167. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/incremental_ridge.py +407 -0
  168. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/linear.py +363 -0
  169. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/logistic_regression.py +466 -0
  170. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/ridge.py +407 -0
  171. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_linear.py +267 -0
  172. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_ridge.py +214 -0
  173. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_linear.py +167 -0
  174. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_logreg.py +565 -0
  175. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_ridge.py +256 -0
  176. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/manifold/t_sne.py +11 -0
  177. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/manifold/tests/test_tsne.py +256 -0
  178. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/metrics/pairwise.py +3 -0
  179. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/metrics/ranking.py +3 -0
  180. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/model_selection/split.py +3 -0
  181. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/_lof.py +30 -62
  182. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/common.py +56 -9
  183. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/knn_classification.py +45 -101
  184. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/knn_regression.py +63 -94
  185. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/knn_unsupervised.py +49 -25
  186. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/tests/test_neighbors.py +6 -4
  187. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/preview/__init__.py +1 -1
  188. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/preview/covariance/covariance.py +261 -0
  189. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/preview/covariance/tests/test_covariance.py +54 -8
  190. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/__init__.py +19 -0
  191. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/incremental_pca.py +406 -0
  192. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/tests/test_incremental_pca.py +390 -0
  193. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/__init__.py +20 -0
  194. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/basic_statistics.py +24 -0
  195. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/incremental_basic_statistics.py +30 -0
  196. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_basic_statistics_spmd.py +117 -0
  197. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_incremental_basic_statistics_spmd.py +314 -0
  198. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/basic_statistics.py → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/cluster/dbscan.py +9 -4
  199. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_dbscan_spmd.py +108 -0
  200. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_kmeans_spmd.py +180 -0
  201. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/covariance.py → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/__init__.py +3 -4
  202. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/covariance.py +24 -0
  203. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/incremental_covariance.py +29 -0
  204. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_covariance_spmd.py +120 -0
  205. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_incremental_covariance_spmd.py +200 -0
  206. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/__init__.py +20 -0
  207. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/incremental_pca.py +23 -0
  208. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/spmd/decomposition/pca.py +6 -4
  209. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_incremental_pca_spmd.py +276 -0
  210. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_pca_spmd.py +146 -0
  211. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/ensemble/forest.py +299 -0
  212. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/ensemble/tests/test_forest_spmd.py +299 -0
  213. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/__init__.py +2 -1
  214. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/incremental_linear_model.py +28 -0
  215. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/linear_model.py +7 -4
  216. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_incremental_linear_spmd.py +345 -0
  217. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_linear_regression_spmd.py +162 -0
  218. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py +169 -0
  219. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd/neighbors/neighbors.py → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/neighbors/__init__.py +1 -3
  220. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/neighbors/tests/test_neighbors_spmd.py +433 -0
  221. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/svm/_common.py +403 -0
  222. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/nusvc.py +99 -117
  223. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/nusvr.py +55 -16
  224. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/svc.py +95 -113
  225. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/svr.py +51 -16
  226. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/tests/test_svm.py +43 -20
  227. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_common.py +607 -0
  228. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_config.py +256 -0
  229. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_hyperparameters.py +111 -0
  230. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_memory_usage.py +335 -0
  231. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/tests/test_monkeypatch.py +5 -4
  232. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_n_jobs_support.py +111 -0
  233. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/tests/test_patching.py +122 -75
  234. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_run_to_run_stability.py +335 -0
  235. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/utils/__init__.py +48 -0
  236. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/utils/base.py +420 -0
  237. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/utils/spmd.py +198 -0
  238. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/utils/validation.py → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/__init__.py +3 -1
  239. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/_array_api.py +217 -0
  240. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/class_weight.py +100 -0
  241. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/parallel.py +97 -0
  242. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/tests/test_class_weight.py +69 -0
  243. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/tests/test_validation.py +238 -0
  244. scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/validation.py +212 -0
  245. scikit_learn_intelex-2025.10.0.dist-info/METADATA +182 -0
  246. scikit_learn_intelex-2025.10.0.dist-info/RECORD +267 -0
  247. {scikit_learn_intelex-2024.4.0.dist-info → scikit_learn_intelex-2025.10.0.dist-info}/WHEEL +1 -1
  248. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/_config.py +0 -110
  249. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/_device_offload.py +0 -250
  250. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/_utils.py +0 -109
  251. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/cluster/k_means.py +0 -17
  252. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/cluster/tests/test_kmeans.py +0 -30
  253. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/covariance/incremental_covariance.py +0 -130
  254. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/covariance/tests/test_incremental_covariance.py +0 -143
  255. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/decomposition/pca.py +0 -335
  256. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/decomposition/tests/test_pca.py +0 -56
  257. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/ensemble/tests/test_forest.py +0 -113
  258. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/linear.py +0 -316
  259. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/logistic_path.py +0 -17
  260. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/logistic_regression.py +0 -385
  261. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_linear.py +0 -117
  262. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_logreg.py +0 -91
  263. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/manifold/tests/test_tsne.py +0 -26
  264. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/preview/cluster/_common.py +0 -84
  265. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/preview/cluster/k_means.py +0 -303
  266. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/preview/covariance/covariance.py +0 -133
  267. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd/cluster/dbscan.py +0 -50
  268. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd/ensemble/forest.py +0 -71
  269. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/svm/_common.py +0 -185
  270. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/_utils.py +0 -164
  271. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/test_config.py +0 -39
  272. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/test_memory_usage.py +0 -227
  273. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/test_n_jobs_support.py +0 -99
  274. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/test_run_to_run_stability_tests.py +0 -428
  275. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/utils/__init__.py +0 -20
  276. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/utils/_namespace.py +0 -97
  277. scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/utils/parallel.py +0 -59
  278. scikit_learn_intelex-2024.4.0.dist-info/METADATA +0 -230
  279. scikit_learn_intelex-2024.4.0.dist-info/RECORD +0 -101
  280. {scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex → scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal}/basic_statistics/__init__.py +0 -0
  281. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/covariance/__init__.py +0 -0
  282. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/decomposition/__init__.py +0 -0
  283. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/ensemble/__init__.py +0 -0
  284. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/manifold/__init__.py +0 -0
  285. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/metrics/__init__.py +0 -0
  286. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/metrics/tests/test_metrics.py +0 -0
  287. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/model_selection/__init__.py +0 -0
  288. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/model_selection/tests/test_model_selection.py +0 -0
  289. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/__init__.py +0 -0
  290. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/preview/covariance/__init__.py +0 -0
  291. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/spmd/__init__.py +0 -0
  292. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/spmd/cluster/__init__.py +0 -0
  293. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/spmd/cluster/kmeans.py +0 -0
  294. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/spmd/ensemble/__init__.py +0 -0
  295. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/logistic_regression.py +0 -0
  296. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/__init__.py +0 -0
  297. {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/tests/test_parallel.py +0 -0
  298. {scikit_learn_intelex-2024.4.0.dist-info → scikit_learn_intelex-2025.10.0.dist-info}/LICENSE.txt +0 -0
  299. {scikit_learn_intelex-2024.4.0.dist-info → scikit_learn_intelex-2025.10.0.dist-info}/top_level.txt +0 -0
@@ -1,99 +0,0 @@
1
- # ==============================================================================
2
- # Copyright 2023 Intel Corporation
3
- #
4
- # Licensed under the Apache License, Version 2.0 (the "License");
5
- # you may not use this file except in compliance with the License.
6
- # You may obtain a copy of the License at
7
- #
8
- # http://www.apache.org/licenses/LICENSE-2.0
9
- #
10
- # Unless required by applicable law or agreed to in writing, software
11
- # distributed under the License is distributed on an "AS IS" BASIS,
12
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
- # See the License for the specific language governing permissions and
14
- # limitations under the License.
15
- # ==============================================================================
16
-
17
- import inspect
18
- import logging
19
- from multiprocessing import cpu_count
20
-
21
- import pytest
22
- from sklearn.base import BaseEstimator
23
- from sklearn.datasets import make_classification
24
-
25
- from sklearnex.dispatcher import get_patch_map
26
- from sklearnex.svm import SVC, NuSVC
27
-
28
- ESTIMATORS = set(
29
- filter(
30
- lambda x: inspect.isclass(x) and issubclass(x, BaseEstimator),
31
- [value[0][0][2] for value in get_patch_map().values()],
32
- )
33
- )
34
-
35
- X, Y = make_classification(n_samples=40, n_features=4, random_state=42)
36
-
37
-
38
- @pytest.mark.parametrize("estimator_class", ESTIMATORS)
39
- @pytest.mark.parametrize("n_jobs", [None, -1, 1, 2])
40
- def test_n_jobs_support(caplog, estimator_class, n_jobs):
41
- def check_estimator_doc(estimator):
42
- if estimator.__doc__ is not None:
43
- assert "n_jobs" in estimator.__doc__
44
-
45
- def check_n_jobs_entry_in_logs(caplog, function_name, n_jobs):
46
- for rec in caplog.records:
47
- if function_name in rec.message and "threads" in rec.message:
48
- expected_n_jobs = n_jobs if n_jobs > 0 else cpu_count() + 1 + n_jobs
49
- logging.info(f"{function_name}: setting {expected_n_jobs} threads")
50
- if f"{function_name}: setting {expected_n_jobs} threads" in rec.message:
51
- return True
52
- # False if n_jobs is set and not found in logs
53
- return n_jobs is None
54
-
55
- def check_method(*args, method, caplog):
56
- method(*args)
57
- assert check_n_jobs_entry_in_logs(caplog, method.__name__, n_jobs)
58
-
59
- def check_methods_decoration(estimator):
60
- funcs = {
61
- i: getattr(estimator, i)
62
- for i in dir(estimator)
63
- if hasattr(estimator, i) and callable(getattr(estimator, i))
64
- }
65
-
66
- for func_name, func in funcs.items():
67
- assert hasattr(func, "__onedal_n_jobs_decorated__") == (
68
- func_name in estimator._n_jobs_supported_onedal_methods
69
- ), f"{estimator}.{func_name} n_jobs decoration does not match {estimator} n_jobs supported methods"
70
-
71
- caplog.set_level(logging.DEBUG, logger="sklearnex")
72
- estimator_kwargs = {"n_jobs": n_jobs}
73
- # by default, [Nu]SVC.predict_proba is restricted by @available_if decorator
74
- if estimator_class in [SVC, NuSVC]:
75
- estimator_kwargs["probability"] = True
76
- estimator_instance = estimator_class(**estimator_kwargs)
77
- # check `n_jobs` parameter doc entry
78
- check_estimator_doc(estimator_class)
79
- check_estimator_doc(estimator_instance)
80
- # check `n_jobs` log entry for supported methods
81
- # `fit` call is required before other methods
82
- check_method(X, Y, method=estimator_instance.fit, caplog=caplog)
83
- for method_name in estimator_instance._n_jobs_supported_onedal_methods:
84
- if method_name == "fit":
85
- continue
86
- method = getattr(estimator_instance, method_name)
87
- argdict = inspect.signature(method).parameters
88
- argnum = len(
89
- [i for i in argdict if argdict[i].default == inspect.Parameter.empty]
90
- )
91
- if argnum == 0:
92
- check_method(method=method, caplog=caplog)
93
- elif argnum == 1:
94
- check_method(X, method=method, caplog=caplog)
95
- else:
96
- check_method(X, Y, method=method, caplog=caplog)
97
- # check if correct methods were decorated
98
- check_methods_decoration(estimator_class)
99
- check_methods_decoration(estimator_instance)
@@ -1,428 +0,0 @@
1
- # ===============================================================================
2
- # Copyright 2020 Intel Corporation
3
- #
4
- # Licensed under the Apache License, Version 2.0 (the "License");
5
- # you may not use this file except in compliance with the License.
6
- # You may obtain a copy of the License at
7
- #
8
- # http://www.apache.org/licenses/LICENSE-2.0
9
- #
10
- # Unless required by applicable law or agreed to in writing, software
11
- # distributed under the License is distributed on an "AS IS" BASIS,
12
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
- # See the License for the specific language governing permissions and
14
- # limitations under the License.
15
- # ===============================================================================
16
-
17
- import random
18
-
19
- import numpy as np
20
- import pytest
21
-
22
- import daal4py as d4p
23
- from sklearnex import patch_sklearn
24
-
25
- patch_sklearn()
26
-
27
- from scipy import sparse
28
- from sklearn.cluster import DBSCAN, KMeans
29
- from sklearn.datasets import (
30
- load_breast_cancer,
31
- load_diabetes,
32
- load_iris,
33
- make_classification,
34
- make_regression,
35
- )
36
- from sklearn.decomposition import PCA
37
- from sklearn.ensemble import RandomForestClassifier, RandomForestRegressor
38
- from sklearn.linear_model import (
39
- ElasticNet,
40
- Lasso,
41
- LinearRegression,
42
- LogisticRegression,
43
- LogisticRegressionCV,
44
- Ridge,
45
- )
46
- from sklearn.manifold import TSNE
47
- from sklearn.metrics import pairwise_distances, roc_auc_score
48
- from sklearn.model_selection import train_test_split
49
- from sklearn.neighbors import (
50
- KNeighborsClassifier,
51
- KNeighborsRegressor,
52
- LocalOutlierFactor,
53
- NearestNeighbors,
54
- )
55
- from sklearn.svm import SVC, SVR, NuSVC, NuSVR
56
-
57
- from daal4py.sklearn._utils import daal_check_version
58
-
59
- # to reproduce errors even in CI
60
- d4p.daalinit(nthreads=100)
61
-
62
-
63
- def get_class_name(x):
64
- return x.__class__.__name__
65
-
66
-
67
- def method_processing(X, clf, methods):
68
- res = []
69
- name = []
70
- for i in methods:
71
- if i == "predict":
72
- res.append(clf.predict(X))
73
- name.append(get_class_name(clf) + ".predict(X)")
74
- elif i == "predict_proba":
75
- res.append(clf.predict_proba(X))
76
- name.append(get_class_name(clf) + ".predict_proba(X)")
77
- elif i == "decision_function":
78
- res.append(clf.decision_function(X))
79
- name.append(get_class_name(clf) + ".decision_function(X)")
80
- elif i == "kneighbors":
81
- dist, idx = clf.kneighbors(X)
82
- res.append(dist)
83
- name.append("dist")
84
- res.append(idx)
85
- name.append("idx")
86
- elif i == "fit_predict":
87
- predict = clf.fit_predict(X)
88
- res.append(predict)
89
- name.append(get_class_name(clf) + ".fit_predict")
90
- elif i == "fit_transform":
91
- res.append(clf.fit_transform(X))
92
- name.append(get_class_name(clf) + ".fit_transform")
93
- elif i == "transform":
94
- res.append(clf.transform(X))
95
- name.append(get_class_name(clf) + ".transform(X)")
96
- elif i == "get_covariance":
97
- res.append(clf.get_covariance())
98
- name.append(get_class_name(clf) + ".get_covariance()")
99
- elif i == "get_precision":
100
- res.append(clf.get_precision())
101
- name.append(get_class_name(clf) + ".get_precision()")
102
- elif i == "score_samples":
103
- res.append(clf.score_samples(X))
104
- name.append(get_class_name(clf) + ".score_samples(X)")
105
- return res, name
106
-
107
-
108
- def func(X, Y, clf, methods):
109
- clf.fit(X, Y)
110
- res, name = method_processing(X, clf, methods)
111
-
112
- for i in clf.__dict__.keys():
113
- ans = getattr(clf, i)
114
- if isinstance(ans, (bool, float, int, np.ndarray, np.float64)):
115
- if isinstance(ans, np.ndarray) and None in ans:
116
- continue
117
- res.append(ans)
118
- name.append(get_class_name(clf) + "." + i)
119
- return res, name
120
-
121
-
122
- def _run_test(model, methods, dataset):
123
- datasets = []
124
- if dataset in ["blobs", "classifier", "sparse"]:
125
- X1, y1 = load_iris(return_X_y=True)
126
- if dataset == "sparse":
127
- X1 = sparse.csr_matrix(X1)
128
- datasets.append((X1, y1))
129
- X2, y2 = load_breast_cancer(return_X_y=True)
130
- if dataset == "sparse":
131
- X2 = sparse.csr_matrix(X2)
132
- datasets.append((X2, y2))
133
- elif dataset == "regression":
134
- X1, y1 = make_regression(
135
- n_samples=500, n_features=10, noise=64.0, random_state=42
136
- )
137
- datasets.append((X1, y1))
138
- X2, y2 = load_diabetes(return_X_y=True)
139
- datasets.append((X2, y2))
140
- else:
141
- raise ValueError("Unknown dataset type")
142
-
143
- for X, y in datasets:
144
- baseline, name = func(X, y, model, methods)
145
- for i in range(10):
146
- res, _ = func(X, y, model, methods)
147
-
148
- for a, b, n in zip(res, baseline, name):
149
- np.testing.assert_allclose(
150
- a, b, rtol=0.0, atol=0.0, err_msg=str(n + " is incorrect")
151
- )
152
-
153
-
154
- MODELS_INFO = [
155
- {
156
- "model": KNeighborsClassifier(
157
- n_neighbors=10, algorithm="brute", weights="uniform"
158
- ),
159
- "methods": ["predict", "predict_proba", "kneighbors"],
160
- "dataset": "classifier",
161
- },
162
- {
163
- "model": KNeighborsClassifier(
164
- n_neighbors=10, algorithm="brute", weights="distance"
165
- ),
166
- "methods": ["predict", "predict_proba", "kneighbors"],
167
- "dataset": "classifier",
168
- },
169
- {
170
- "model": KNeighborsClassifier(
171
- n_neighbors=10, algorithm="kd_tree", weights="uniform"
172
- ),
173
- "methods": ["predict", "predict_proba", "kneighbors"],
174
- "dataset": "classifier",
175
- },
176
- {
177
- "model": KNeighborsClassifier(
178
- n_neighbors=10, algorithm="kd_tree", weights="distance"
179
- ),
180
- "methods": ["predict", "predict_proba", "kneighbors"],
181
- "dataset": "classifier",
182
- },
183
- {
184
- "model": KNeighborsRegressor(
185
- n_neighbors=10, algorithm="kd_tree", weights="distance"
186
- ),
187
- "methods": ["predict", "kneighbors"],
188
- "dataset": "regression",
189
- },
190
- {
191
- "model": KNeighborsRegressor(
192
- n_neighbors=10, algorithm="kd_tree", weights="uniform"
193
- ),
194
- "methods": ["predict", "kneighbors"],
195
- "dataset": "regression",
196
- },
197
- {
198
- "model": KNeighborsRegressor(
199
- n_neighbors=10, algorithm="brute", weights="distance"
200
- ),
201
- "methods": ["predict", "kneighbors"],
202
- "dataset": "regression",
203
- },
204
- {
205
- "model": KNeighborsRegressor(
206
- n_neighbors=10, algorithm="brute", weights="uniform"
207
- ),
208
- "methods": ["predict", "kneighbors"],
209
- "dataset": "regression",
210
- },
211
- {
212
- "model": NearestNeighbors(n_neighbors=10, algorithm="brute"),
213
- "methods": ["kneighbors"],
214
- "dataset": "blobs",
215
- },
216
- {
217
- "model": NearestNeighbors(n_neighbors=10, algorithm="kd_tree"),
218
- "methods": ["kneighbors"],
219
- "dataset": "blobs",
220
- },
221
- {
222
- "model": LocalOutlierFactor(n_neighbors=10, novelty=False),
223
- "methods": ["fit_predict"],
224
- "dataset": "blobs",
225
- },
226
- {
227
- "model": LocalOutlierFactor(n_neighbors=10, novelty=True),
228
- "methods": ["predict"],
229
- "dataset": "blobs",
230
- },
231
- {
232
- "model": DBSCAN(algorithm="brute", n_jobs=-1),
233
- "methods": [],
234
- "dataset": "blobs",
235
- },
236
- {
237
- "model": SVC(kernel="rbf"),
238
- "methods": ["predict", "decision_function"],
239
- "dataset": "classifier",
240
- },
241
- {
242
- "model": SVC(kernel="rbf"),
243
- "methods": ["predict", "decision_function"],
244
- "dataset": "sparse",
245
- },
246
- {
247
- "model": NuSVC(kernel="rbf"),
248
- "methods": ["predict", "decision_function"],
249
- "dataset": "classifier",
250
- },
251
- {
252
- "model": SVR(kernel="rbf"),
253
- "methods": ["predict"],
254
- "dataset": "regression",
255
- },
256
- {
257
- "model": NuSVR(kernel="rbf"),
258
- "methods": ["predict"],
259
- "dataset": "regression",
260
- },
261
- {
262
- "model": TSNE(random_state=0),
263
- "methods": ["fit_transform"],
264
- "dataset": "classifier",
265
- },
266
- {
267
- "model": KMeans(random_state=0, init="k-means++"),
268
- "methods": ["predict"],
269
- "dataset": "blobs",
270
- },
271
- {
272
- "model": KMeans(random_state=0, init="random"),
273
- "methods": ["predict"],
274
- "dataset": "blobs",
275
- },
276
- {
277
- "model": KMeans(random_state=0, init="k-means++"),
278
- "methods": ["predict"],
279
- "dataset": "sparse",
280
- },
281
- {
282
- "model": KMeans(random_state=0, init="random"),
283
- "methods": ["predict"],
284
- "dataset": "sparse",
285
- },
286
- {
287
- "model": ElasticNet(random_state=0),
288
- "methods": ["predict"],
289
- "dataset": "regression",
290
- },
291
- {
292
- "model": Lasso(random_state=0),
293
- "methods": ["predict"],
294
- "dataset": "regression",
295
- },
296
- {
297
- "model": PCA(n_components=0.5, svd_solver="full", random_state=0),
298
- "methods": ["transform", "get_covariance", "get_precision", "score_samples"],
299
- "dataset": "classifier",
300
- },
301
- {
302
- "model": RandomForestClassifier(
303
- random_state=0, oob_score=True, max_samples=0.5, max_features="sqrt"
304
- ),
305
- "methods": ["predict", "predict_proba"],
306
- "dataset": "classifier",
307
- },
308
- {
309
- "model": LogisticRegression(random_state=0, solver="newton-cg", max_iter=1000),
310
- "methods": ["predict", "predict_proba"],
311
- "dataset": "classifier",
312
- },
313
- {
314
- "model": LogisticRegression(random_state=0, solver="lbfgs", max_iter=1000),
315
- "methods": ["predict", "predict_proba"],
316
- "dataset": "classifier",
317
- },
318
- {
319
- "model": LogisticRegressionCV(
320
- random_state=0, solver="newton-cg", n_jobs=-1, max_iter=1000
321
- ),
322
- "methods": ["predict", "predict_proba"],
323
- "dataset": "classifier",
324
- },
325
- {
326
- "model": LogisticRegressionCV(
327
- random_state=0, solver="lbfgs", n_jobs=-1, max_iter=1000
328
- ),
329
- "methods": ["predict", "predict_proba"],
330
- "dataset": "classifier",
331
- },
332
- {
333
- "model": RandomForestRegressor(
334
- random_state=0, oob_score=True, max_samples=0.5, max_features="sqrt"
335
- ),
336
- "methods": ["predict"],
337
- "dataset": "regression",
338
- },
339
- {
340
- "model": LinearRegression(),
341
- "methods": ["predict"],
342
- "dataset": "regression",
343
- },
344
- {
345
- "model": Ridge(random_state=0),
346
- "methods": ["predict"],
347
- "dataset": "regression",
348
- },
349
- ]
350
-
351
- TO_SKIP = [
352
- "TSNE", # Absolute diff is 1e-10, potential problem in KNN,
353
- # will be fixed for next release. (UPD. KNN is fixed but there is a problem
354
- # with stability of stock sklearn. It is already stable in master, so, we
355
- # need to wait for the next sklearn release)
356
- "LogisticRegression", # Absolute diff is 1e-8, will be fixed for next release
357
- "LogisticRegressionCV", # Absolute diff is 1e-10, will be fixed for next release
358
- "RandomForestRegressor", # Absolute diff is 1e-14 in OOB score,
359
- # will be fixed for next release
360
- ]
361
-
362
-
363
- @pytest.mark.parametrize("model_head", MODELS_INFO)
364
- def test_models(model_head):
365
- stable_algos = []
366
- if get_class_name(model_head["model"]) in stable_algos and daal_check_version(
367
- (2021, "P", 300)
368
- ):
369
- try:
370
- TO_SKIP.remove(get_class_name(model_head["model"]))
371
- except ValueError:
372
- pass
373
- if get_class_name(model_head["model"]) in TO_SKIP:
374
- pytest.skip("Unstable", allow_module_level=False)
375
- _run_test(model_head["model"], model_head["methods"], model_head["dataset"])
376
-
377
-
378
- @pytest.mark.parametrize("features", range(5, 10))
379
- def test_train_test_split(features):
380
- X, y = make_classification(
381
- n_samples=4000,
382
- n_features=features,
383
- n_informative=features,
384
- n_redundant=0,
385
- n_clusters_per_class=8,
386
- random_state=0,
387
- )
388
- (
389
- baseline_X_train,
390
- baseline_X_test,
391
- baseline_y_train,
392
- baseline_y_test,
393
- ) = train_test_split(X, y, test_size=0.33, random_state=0)
394
- baseline = [baseline_X_train, baseline_X_test, baseline_y_train, baseline_y_test]
395
- for _ in range(10):
396
- X_train, X_test, y_train, y_test = train_test_split(
397
- X, y, test_size=0.33, random_state=0
398
- )
399
- res = [X_train, X_test, y_train, y_test]
400
- for a, b in zip(res, baseline):
401
- np.testing.assert_allclose(
402
- a, b, rtol=0.0, atol=0.0, err_msg=str("train_test_split is incorrect")
403
- )
404
-
405
-
406
- @pytest.mark.parametrize("metric", ["cosine", "correlation"])
407
- def test_pairwise_distances(metric):
408
- X = np.random.rand(1000)
409
- X = np.array(X, dtype=np.float64)
410
- baseline = pairwise_distances(X.reshape(1, -1), metric=metric)
411
- for _ in range(5):
412
- res = pairwise_distances(X.reshape(1, -1), metric=metric)
413
- for a, b in zip(res, baseline):
414
- np.testing.assert_allclose(
415
- a, b, rtol=0.0, atol=0.0, err_msg=str("pairwise_distances is incorrect")
416
- )
417
-
418
-
419
- @pytest.mark.parametrize("array_size", [100, 1000, 10000])
420
- def test_roc_auc(array_size):
421
- a = [random.randint(0, 1) for i in range(array_size)]
422
- b = [random.randint(0, 1) for i in range(array_size)]
423
- baseline = roc_auc_score(a, b)
424
- for _ in range(5):
425
- res = roc_auc_score(a, b)
426
- np.testing.assert_allclose(
427
- baseline, res, rtol=0.0, atol=0.0, err_msg=str("roc_auc is incorrect")
428
- )
@@ -1,20 +0,0 @@
1
- # ===============================================================================
2
- # Copyright 2022 Intel Corporation
3
- #
4
- # Licensed under the Apache License, Version 2.0 (the "License");
5
- # you may not use this file except in compliance with the License.
6
- # You may obtain a copy of the License at
7
- #
8
- # http://www.apache.org/licenses/LICENSE-2.0
9
- #
10
- # Unless required by applicable law or agreed to in writing, software
11
- # distributed under the License is distributed on an "AS IS" BASIS,
12
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
- # See the License for the specific language governing permissions and
14
- # limitations under the License.
15
- # ===============================================================================
16
-
17
- from ._namespace import get_namespace
18
- from .validation import _assert_all_finite
19
-
20
- __all__ = ["get_namespace", "_assert_all_finite"]
@@ -1,97 +0,0 @@
1
- # ==============================================================================
2
- # Copyright 2024 Intel Corporation
3
- #
4
- # Licensed under the Apache License, Version 2.0 (the "License");
5
- # you may not use this file except in compliance with the License.
6
- # You may obtain a copy of the License at
7
- #
8
- # http://www.apache.org/licenses/LICENSE-2.0
9
- #
10
- # Unless required by applicable law or agreed to in writing, software
11
- # distributed under the License is distributed on an "AS IS" BASIS,
12
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
- # See the License for the specific language governing permissions and
14
- # limitations under the License.
15
- # ==============================================================================
16
-
17
- import numpy as np
18
-
19
- from daal4py.sklearn._utils import sklearn_check_version
20
-
21
- from .._device_offload import dpnp_available
22
-
23
- if sklearn_check_version("1.2"):
24
- from sklearn.utils._array_api import get_namespace as sklearn_get_namespace
25
-
26
- if dpnp_available:
27
- import dpnp
28
-
29
-
30
- def get_namespace(*arrays):
31
- """Get namespace of arrays.
32
-
33
- Introspect `arrays` arguments and return their common Array API
34
- compatible namespace object, if any. NumPy 1.22 and later can
35
- construct such containers using the `numpy.array_api` namespace
36
- for instance.
37
-
38
- This function will return the namespace of SYCL-related arrays
39
- which define the __sycl_usm_array_interface__ attribute
40
- regardless of array_api support, the configuration of
41
- array_api_dispatch, or scikit-learn version.
42
-
43
- See: https://numpy.org/neps/nep-0047-array-api-standard.html
44
-
45
- If `arrays` are regular numpy arrays, an instance of the
46
- `_NumPyApiWrapper` compatibility wrapper is returned instead.
47
-
48
- Namespace support is not enabled by default. To enabled it
49
- call:
50
-
51
- sklearn.set_config(array_api_dispatch=True)
52
-
53
- or:
54
-
55
- with sklearn.config_context(array_api_dispatch=True):
56
- # your code here
57
-
58
- Otherwise an instance of the `_NumPyApiWrapper`
59
- compatibility wrapper is always returned irrespective of
60
- the fact that arrays implement the `__array_namespace__`
61
- protocol or not.
62
-
63
- Parameters
64
- ----------
65
- *arrays : array objects
66
- Array objects.
67
-
68
- Returns
69
- -------
70
- namespace : module
71
- Namespace shared by array objects.
72
-
73
- is_array_api : bool
74
- True of the arrays are containers that implement the Array API spec.
75
- """
76
-
77
- # sycl support designed to work regardless of array_api_dispatch sklearn global value
78
- sycl_type = {type(x): x for x in arrays if hasattr(x, "__sycl_usm_array_interface__")}
79
-
80
- if len(sycl_type) > 1:
81
- raise ValueError(f"Multiple SYCL types for array inputs: {sycl_type}")
82
-
83
- if sycl_type:
84
-
85
- (X,) = sycl_type.values()
86
-
87
- if hasattr(X, "__array_namespace__"):
88
- return X.__array_namespace__(), True
89
- elif dpnp_available and isinstance(X, dpnp.ndarray):
90
- return dpnp, False
91
- else:
92
- raise ValueError(f"SYCL type not recognized: {sycl_type}")
93
-
94
- elif sklearn_check_version("1.2"):
95
- return sklearn_get_namespace(*arrays)
96
- else:
97
- return np, True