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.
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/__init__.py +73 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/__main__.py +58 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/_daal4py.cp312-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mb/__init__.py +94 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mb/gbt_convertors.py +1199 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mb/logistic_regression_builders.py +211 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mb/tree_based_builders.py +425 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mpi_transceiver.cp312-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/__init__.py +40 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/_n_jobs_support.py +252 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/_utils.py +245 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/__init__.py +20 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/dbscan.py +165 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/k_means.py +565 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/tests/test_dbscan.py +109 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/decomposition/_pca.py +528 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/AdaBoostClassifier.py +192 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/GBTDAAL.py +333 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/__init__.py +27 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/_forest.py +1285 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/tests/test_decision_forest.py +206 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_coordinate_descent.py +826 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_linear.py +269 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_ridge.py +290 -0
- 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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/linear.py +17 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/logistic_loss.py +195 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/logistic_path.py +561 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/ridge.py +17 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_enet.py +157 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_linear.py +198 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_ridge.py +69 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/manifold/_t_sne.py +432 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/metrics/__init__.py +20 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/metrics/_pairwise.py +259 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/metrics/_ranking.py +210 -0
- 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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/model_selection/_split.py +309 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/model_selection/tests/test_split.py +56 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/__init__.py +0 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/dispatcher.py +232 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/_models_info.py +164 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/test_monkeypatch.py +71 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/test_patching.py +90 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/utils/_launch_algorithms.py +117 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/__init__.py +21 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_base.py +493 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_classification.py +136 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_regression.py +72 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_unsupervised.py +55 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/tests/test_kneighbors.py +113 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/svm/__init__.py +19 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/svm/svm.py +736 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/utils/base.py +80 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/utils/tests/test_utils.py +51 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/sklearn/utils/validation.py +772 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/__init__.py +151 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/_config.py +80 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/_device_offload.py +199 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/_onedal_py_dpc.cp312-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/_onedal_py_host.cp312-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/basic_statistics.py +151 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/incremental_basic_statistics.py +165 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/tests/test_basic_statistics.py +241 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/tests/test_incremental_basic_statistics.py +279 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/basic_statistics/tests/utils.py +50 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/__init__.py +27 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/dbscan.py +80 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/kmeans.py +582 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/kmeans_init.py +145 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/tests/test_dbscan.py +125 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/tests/test_kmeans.py +88 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/cluster/tests/test_kmeans_init.py +93 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/_backend.py +258 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/_estimator_checks.py +47 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/_mixin.py +62 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/hyperparameters.py +148 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/common/tests/test_sycl.py +148 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/__init__.py +20 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/covariance.py +121 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/incremental_covariance.py +151 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/tests/test_covariance.py +50 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/covariance/tests/test_incremental_covariance.py +190 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/__init__.py +29 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/_data_conversion.py +158 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/_dlpack.py +64 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/_sycl_usm.py +63 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/tests/common.py +131 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/datatypes/tests/test_data.py +686 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/decomposition/incremental_pca.py +218 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/decomposition/pca.py +124 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/decomposition/tests/test_incremental_pca.py +291 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/dummy/__init__.py +19 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/dummy/dummy.py +137 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/ensemble/__init__.py +29 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/ensemble/forest.py +781 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/ensemble/tests/test_random_forest.py +97 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/__init__.py +27 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/incremental_linear_model.py +201 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/linear_model.py +230 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/logistic_regression.py +293 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_incremental_linear_regression.py +213 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_incremental_ridge_regression.py +171 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_linear_regression.py +252 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_logistic_regression.py +95 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_ridge.py +95 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/neighbors/neighbors.py +690 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/neighbors/tests/test_knn_classification.py +49 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/primitives/__init__.py +27 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/primitives/get_tree.py +27 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/primitives/kernel_functions.py +202 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/primitives/tests/test_kernel_functions.py +159 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/__init__.py +19 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/svm.py +592 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_csr_svm.py +352 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_nusvc.py +204 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_nusvr.py +210 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_svc.py +168 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/svm/tests/test_svr.py +243 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/tests/test_common.py +71 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/tests/utils/_dataframes_support.py +179 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/tests/utils/_device_selection.py +94 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/_array_api.py +98 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/_sycl_queue_manager.py +213 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/_third_party.py +220 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/tests/test_validation.py +142 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/onedal/utils/validation.py +503 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/__init__.py +7 -3
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/__main__.py +2 -2
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/_config.py +163 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/_device_offload.py +205 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/_utils.py +219 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/base.py +109 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/basic_statistics/__init__.py +20 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/basic_statistics/basic_statistics.py +241 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/basic_statistics/tests/test_basic_statistics.py +405 -0
- {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
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/cluster/__init__.py +1 -1
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/cluster/dbscan.py +38 -29
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/cluster/k_means.py +399 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/cluster/tests/test_kmeans.py +157 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/conftest.py +20 -1
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/covariance/incremental_covariance.py +440 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/covariance/tests/test_incremental_covariance.py +307 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/decomposition/pca.py +558 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/decomposition/tests/test_pca.py +164 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/dispatcher.py +199 -21
- {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
- 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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/dummy/_dummy.py +615 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/dummy/tests/test_dummy.py +62 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/ensemble/_forest.py +288 -440
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/ensemble/tests/test_forest.py +196 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/glob/__main__.py +1 -1
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/glob/dispatcher.py +17 -3
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/__init__.py +32 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/coordinate_descent.py +44 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/incremental_linear.py +427 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/incremental_ridge.py +407 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/linear.py +363 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/logistic_regression.py +466 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/ridge.py +407 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_linear.py +267 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_ridge.py +214 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_linear.py +167 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_logreg.py +565 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_ridge.py +256 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/manifold/tests/test_tsne.py +256 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/metrics/pairwise.py +3 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/metrics/ranking.py +3 -0
- {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
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/_lof.py +30 -62
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/common.py +56 -9
- {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
- {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
- {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
- {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
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/preview/__init__.py +1 -1
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/preview/covariance/covariance.py +261 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/__init__.py +19 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/incremental_pca.py +406 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/tests/test_incremental_pca.py +390 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/__init__.py +20 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/basic_statistics.py +24 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/incremental_basic_statistics.py +30 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_basic_statistics_spmd.py +117 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_incremental_basic_statistics_spmd.py +314 -0
- 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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_dbscan_spmd.py +108 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_kmeans_spmd.py +180 -0
- 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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/covariance.py +24 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/incremental_covariance.py +29 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_covariance_spmd.py +120 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_incremental_covariance_spmd.py +200 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/__init__.py +20 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/incremental_pca.py +23 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_incremental_pca_spmd.py +276 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_pca_spmd.py +146 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/ensemble/forest.py +299 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/ensemble/tests/test_forest_spmd.py +299 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/incremental_linear_model.py +28 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_incremental_linear_spmd.py +345 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_linear_regression_spmd.py +162 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py +169 -0
- 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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/spmd/neighbors/tests/test_neighbors_spmd.py +433 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/svm/_common.py +403 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/nusvc.py +99 -117
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/nusvr.py +55 -16
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/svc.py +95 -113
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/svr.py +51 -16
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_common.py +607 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_config.py +256 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_hyperparameters.py +111 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_memory_usage.py +335 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_n_jobs_support.py +111 -0
- {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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/test_run_to_run_stability.py +335 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/utils/__init__.py +48 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/utils/base.py +420 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/tests/utils/spmd.py +198 -0
- 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
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/_array_api.py +217 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/class_weight.py +100 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/parallel.py +97 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/tests/test_class_weight.py +69 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/tests/test_validation.py +238 -0
- scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/sklearnex/utils/validation.py +212 -0
- scikit_learn_intelex-2025.10.0.dist-info/METADATA +182 -0
- scikit_learn_intelex-2025.10.0.dist-info/RECORD +267 -0
- {scikit_learn_intelex-2024.4.0.dist-info → scikit_learn_intelex-2025.10.0.dist-info}/WHEEL +1 -1
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/_config.py +0 -110
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/_device_offload.py +0 -250
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/_utils.py +0 -109
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/cluster/k_means.py +0 -17
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/cluster/tests/test_kmeans.py +0 -30
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/covariance/incremental_covariance.py +0 -130
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/covariance/tests/test_incremental_covariance.py +0 -143
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/decomposition/pca.py +0 -335
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/decomposition/tests/test_pca.py +0 -56
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/ensemble/tests/test_forest.py +0 -113
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/linear.py +0 -316
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/logistic_path.py +0 -17
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/logistic_regression.py +0 -385
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_linear.py +0 -117
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_logreg.py +0 -91
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/manifold/tests/test_tsne.py +0 -26
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/preview/cluster/_common.py +0 -84
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/preview/cluster/k_means.py +0 -303
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/preview/covariance/covariance.py +0 -133
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd/cluster/dbscan.py +0 -50
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/spmd/ensemble/forest.py +0 -71
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/svm/_common.py +0 -185
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/_utils.py +0 -164
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/test_config.py +0 -39
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/test_memory_usage.py +0 -227
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/test_n_jobs_support.py +0 -99
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/test_run_to_run_stability_tests.py +0 -428
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/utils/__init__.py +0 -20
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/utils/_namespace.py +0 -97
- scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/utils/parallel.py +0 -59
- scikit_learn_intelex-2024.4.0.dist-info/METADATA +0 -230
- scikit_learn_intelex-2024.4.0.dist-info/RECORD +0 -101
- {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
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/covariance/__init__.py +0 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/decomposition/__init__.py +0 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/ensemble/__init__.py +0 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/manifold/__init__.py +0 -0
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/metrics/__init__.py +0 -0
- {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
- {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
- {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
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/neighbors/__init__.py +0 -0
- {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
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/spmd/__init__.py +0 -0
- {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
- {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
- {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
- {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
- {scikit_learn_intelex-2024.4.0.data → scikit_learn_intelex-2025.10.0.data}/data/Lib/site-packages/sklearnex/svm/__init__.py +0 -0
- {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
- {scikit_learn_intelex-2024.4.0.dist-info → scikit_learn_intelex-2025.10.0.dist-info}/LICENSE.txt +0 -0
- {scikit_learn_intelex-2024.4.0.dist-info → scikit_learn_intelex-2025.10.0.dist-info}/top_level.txt +0 -0
scikit_learn_intelex-2024.4.0.data/data/Lib/site-packages/sklearnex/tests/test_n_jobs_support.py
DELETED
|
@@ -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
|