scikit-learn-intelex 2024.2.0__py39-none-win_amd64.whl → 2025.1.0__py39-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.
Potentially problematic release.
This version of scikit-learn-intelex might be problematic. Click here for more details.
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/__init__.py +73 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/__main__.py +58 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/_daal4py.cp39-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/doc/third-party-programs.txt +424 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/mb/__init__.py +19 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/mb/model_builders.py +377 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/mpi_transceiver.cp39-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/__init__.py +40 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/_n_jobs_support.py +248 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/_utils.py +245 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/__init__.py +20 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/dbscan.py +165 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/k_means.py +597 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/cluster/tests/test_dbscan.py +109 -0
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/spmd → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn}/decomposition/__init__.py +2 -2
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/decomposition/_pca.py +524 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/AdaBoostClassifier.py +196 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/GBTDAAL.py +337 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/__init__.py +27 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/_forest.py +1397 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/ensemble/tests/test_decision_forest.py +206 -0
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn}/linear_model/__init__.py +29 -28
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_coordinate_descent.py +848 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_linear.py +272 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_ridge.py +325 -0
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/basic_statistics/basic_statistics.py → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/coordinate_descent.py +2 -2
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/linear.py +17 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/logistic_loss.py +195 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/logistic_path.py +1026 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/ridge.py +17 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_linear.py +208 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_ridge.py +69 -0
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/preview/cluster → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/manifold}/__init__.py +3 -3
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/manifold/_t_sne.py +405 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/metrics/__init__.py +20 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/metrics/_pairwise.py +236 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/metrics/_ranking.py +210 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/model_selection/__init__.py +19 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/model_selection/_split.py +309 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/model_selection/tests/test_split.py +56 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/__init__.py +0 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/dispatcher.py +232 -0
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch}/tests/_models_info.py +13 -22
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/test_monkeypatch.py +71 -0
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch}/tests/test_patching.py +10 -56
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch}/tests/utils/_launch_algorithms.py +4 -5
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/__init__.py +21 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_base.py +503 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_classification.py +139 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_regression.py +74 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_unsupervised.py +55 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/neighbors/tests/test_kneighbors.py +113 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/svm/__init__.py +19 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/svm/svm.py +734 -0
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/spmd/covariance → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/utils}/__init__.py +5 -3
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/utils/base.py +75 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/utils/tests/test_utils.py +51 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/daal4py/sklearn/utils/validation.py +693 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/__init__.py +83 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/_config.py +54 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/_device_offload.py +222 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/_onedal_py_dpc.cp39-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/_onedal_py_host.cp39-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/basic_statistics/__init__.py +20 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/basic_statistics/basic_statistics.py +107 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/basic_statistics/incremental_basic_statistics.py +160 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/basic_statistics/tests/test_basic_statistics.py +298 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/basic_statistics/tests/test_incremental_basic_statistics.py +196 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/cluster/__init__.py +27 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/cluster/dbscan.py +110 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/cluster/kmeans.py +564 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/cluster/kmeans_init.py +115 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/cluster/tests/test_dbscan.py +125 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/cluster/tests/test_kmeans.py +88 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/cluster/tests/test_kmeans_init.py +93 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/common/_base.py +38 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/common/_estimator_checks.py +47 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/common/_mixin.py +62 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/common/_policy.py +59 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/common/_spmd_policy.py +30 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/common/hyperparameters.py +125 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/common/tests/test_policy.py +76 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/covariance/__init__.py +20 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/covariance/covariance.py +125 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/covariance/incremental_covariance.py +146 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/covariance/tests/test_covariance.py +50 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/covariance/tests/test_incremental_covariance.py +122 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/datatypes/__init__.py +19 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/datatypes/_data_conversion.py +154 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/datatypes/tests/common.py +126 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/datatypes/tests/test_data.py +414 -0
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/basic_statistics → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/decomposition}/__init__.py +3 -2
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/decomposition/incremental_pca.py +204 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/decomposition/pca.py +186 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/decomposition/tests/test_incremental_pca.py +198 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/ensemble/__init__.py +29 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/ensemble/forest.py +727 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/ensemble/tests/test_random_forest.py +97 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/linear_model/__init__.py +27 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/linear_model/incremental_linear_model.py +258 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/linear_model/linear_model.py +329 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/linear_model/logistic_regression.py +249 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_incremental_linear_regression.py +168 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_incremental_ridge_regression.py +107 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_linear_regression.py +250 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_logistic_regression.py +95 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/linear_model/tests/test_ridge.py +95 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/neighbors/__init__.py +19 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/neighbors/neighbors.py +767 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/neighbors/tests/test_knn_classification.py +49 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/primitives/__init__.py +27 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/primitives/get_tree.py +25 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/primitives/kernel_functions.py +153 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/primitives/tests/test_kernel_functions.py +159 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/svm/__init__.py +19 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/svm/svm.py +556 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/svm/tests/test_csr_svm.py +351 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/svm/tests/test_nusvc.py +204 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/svm/tests/test_nusvr.py +210 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/svm/tests/test_svc.py +176 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/svm/tests/test_svr.py +243 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/tests/test_common.py +57 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/tests/utils/_dataframes_support.py +162 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/tests/utils/_device_selection.py +102 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/utils/__init__.py +49 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/utils/_array_api.py +81 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/utils/_dpep_helpers.py +56 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/onedal/utils/validation.py +440 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/__init__.py +10 -7
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/_config.py +22 -16
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/_device_offload.py +126 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/_utils.py +27 -4
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/basic_statistics/__init__.py +20 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/basic_statistics/basic_statistics.py +230 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/basic_statistics/incremental_basic_statistics.py +345 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/basic_statistics/tests/test_basic_statistics.py +270 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/basic_statistics/tests/test_incremental_basic_statistics.py +404 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/cluster/__init__.py +1 -1
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/cluster/dbscan.py +16 -7
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/cluster/k_means.py +395 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/cluster/tests/test_dbscan.py +8 -6
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/cluster/tests/test_kmeans.py +159 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/conftest.py +82 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/covariance/incremental_covariance.py +398 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/covariance/tests/test_incremental_covariance.py +111 -17
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/decomposition/pca.py +425 -0
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/tests/test_preview_pca.py → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/decomposition/tests/test_pca.py +25 -9
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/dispatcher.py +222 -42
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/ensemble/_forest.py +249 -182
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/ensemble/tests/test_forest.py +39 -21
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/glob/dispatcher.py +16 -2
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/__init__.py +32 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/linear_model/coordinate_descent.py +13 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/incremental_linear.py +482 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/incremental_ridge.py +425 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/linear.py +341 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/linear_model/logistic_regression.py +182 -102
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/linear_model/ridge.py +7 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_linear.py +207 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_ridge.py +153 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_linear.py +167 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/linear_model/tests/test_logreg.py +45 -4
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/manifold/t_sne.py +4 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/metrics/pairwise.py +5 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/metrics/ranking.py +3 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/model_selection/split.py +5 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/neighbors/_lof.py +97 -28
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/neighbors/common.py +53 -6
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/neighbors/knn_classification.py +48 -149
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/neighbors/knn_regression.py +43 -144
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/neighbors/knn_unsupervised.py +50 -93
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/neighbors/tests/test_neighbors.py +6 -9
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/preview/__init__.py +1 -1
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/preview/covariance/covariance.py +24 -18
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/preview/covariance/tests/test_covariance.py +18 -5
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/__init__.py +19 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/incremental_pca.py +233 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/tests/test_incremental_pca.py +266 -0
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/preview/decomposition → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/preview/linear_model}/__init__.py +19 -19
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/preview/linear_model/ridge.py +424 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/preview/linear_model/tests/test_ridge.py +102 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/__init__.py +20 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/incremental_basic_statistics.py +30 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_basic_statistics_spmd.py +107 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_incremental_basic_statistics_spmd.py +307 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_dbscan_spmd.py +97 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_kmeans_spmd.py +172 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/__init__.py +20 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/incremental_covariance.py +37 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_covariance_spmd.py +107 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_incremental_covariance_spmd.py +184 -0
- {scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition}/__init__.py +3 -2
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/incremental_pca.py +30 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_incremental_pca_spmd.py +269 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_pca_spmd.py +128 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/ensemble/forest.py +4 -12
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/ensemble/tests/test_forest_spmd.py +265 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/__init__.py +2 -1
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/tests/test_config.py → scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/incremental_linear_model.py +14 -18
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_incremental_linear_spmd.py +329 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_linear_regression_spmd.py +145 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py +162 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/spmd/neighbors/tests/test_neighbors_spmd.py +288 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/svm/_common.py +339 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/svm/nusvc.py +168 -73
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/svm/nusvr.py +71 -66
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/svm/svc.py +166 -72
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/svm/svr.py +64 -63
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/svm/tests/test_svm.py +12 -20
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/test_common.py +390 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/test_config.py +123 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/test_memory_usage.py +379 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/test_monkeypatch.py +276 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/test_n_jobs_support.py +108 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/tests/test_parallel.py +6 -8
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/test_patching.py +385 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/test_run_to_run_stability.py +321 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/utils/__init__.py +44 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/utils/base.py +371 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/tests/utils/spmd.py +198 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/utils/_array_api.py +82 -0
- scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/utils/tests/test_finite.py +89 -0
- {scikit_learn_intelex-2024.2.0.dist-info → scikit_learn_intelex-2025.1.0.dist-info}/METADATA +231 -230
- scikit_learn_intelex-2025.1.0.dist-info/RECORD +257 -0
- {scikit_learn_intelex-2024.2.0.dist-info → scikit_learn_intelex-2025.1.0.dist-info}/WHEEL +1 -1
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/_device_offload.py +0 -223
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/cluster/k_means.py +0 -17
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/cluster/tests/test_kmeans.py +0 -30
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/covariance/incremental_covariance.py +0 -130
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/decomposition/pca.py +0 -17
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/decomposition/tests/test_pca.py +0 -27
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/linear_model/linear.py +0 -381
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/linear_model/logistic_path.py +0 -17
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_linear.py +0 -82
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/preview/cluster/_common.py +0 -84
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/preview/cluster/k_means.py +0 -371
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/preview/decomposition/pca.py +0 -374
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/svm/_common.py +0 -188
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/tests/test_memory_usage.py +0 -222
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/tests/test_monkeypatch.py +0 -240
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/tests/test_n_jobs_support.py +0 -93
- scikit_learn_intelex-2024.2.0.data/data/Lib/site-packages/sklearnex/tests/test_run_to_run_stability_tests.py +0 -428
- scikit_learn_intelex-2024.2.0.dist-info/RECORD +0 -101
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/__main__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/covariance/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/decomposition/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/doc/third-party-programs.txt +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/ensemble/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/glob/__main__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/manifold/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/manifold/tests/test_tsne.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/metrics/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/metrics/tests/test_metrics.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/model_selection/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/model_selection/tests/test_model_selection.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/neighbors/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/preview/covariance/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/basic_statistics/basic_statistics.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/cluster/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/cluster/dbscan.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/cluster/kmeans.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/covariance/covariance.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/decomposition/pca.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/ensemble/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/linear_model.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/logistic_regression.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/neighbors/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/spmd/neighbors/neighbors.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/svm/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/utils/__init__.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/utils/parallel.py +0 -0
- {scikit_learn_intelex-2024.2.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/utils/validation.py +0 -0
- {scikit_learn_intelex-2024.2.0.dist-info → scikit_learn_intelex-2025.1.0.dist-info}/LICENSE.txt +0 -0
- {scikit_learn_intelex-2024.2.0.dist-info → scikit_learn_intelex-2025.1.0.dist-info}/top_level.txt +0 -0
|
@@ -14,92 +14,29 @@
|
|
|
14
14
|
# limitations under the License.
|
|
15
15
|
# ===============================================================================
|
|
16
16
|
|
|
17
|
-
|
|
18
|
-
from packaging.version import Version
|
|
19
|
-
except ImportError:
|
|
20
|
-
from distutils.version import LooseVersion as Version
|
|
21
|
-
|
|
22
|
-
import warnings
|
|
23
|
-
|
|
24
|
-
import numpy as np
|
|
25
|
-
from sklearn import __version__ as sklearn_version
|
|
26
|
-
from sklearn.neighbors._ball_tree import BallTree
|
|
27
|
-
from sklearn.neighbors._base import VALID_METRICS
|
|
28
|
-
from sklearn.neighbors._base import NeighborsBase as sklearn_NeighborsBase
|
|
29
|
-
from sklearn.neighbors._kd_tree import KDTree
|
|
30
|
-
from sklearn.neighbors._unsupervised import NearestNeighbors as sklearn_NearestNeighbors
|
|
17
|
+
from sklearn.neighbors._unsupervised import NearestNeighbors as _sklearn_NearestNeighbors
|
|
31
18
|
from sklearn.utils.validation import _deprecate_positional_args, check_is_fitted
|
|
32
19
|
|
|
33
20
|
from daal4py.sklearn._n_jobs_support import control_n_jobs
|
|
34
21
|
from daal4py.sklearn._utils import sklearn_check_version
|
|
35
22
|
from onedal.neighbors import NearestNeighbors as onedal_NearestNeighbors
|
|
36
|
-
from onedal.utils import _check_array, _num_features, _num_samples
|
|
37
23
|
|
|
38
24
|
from .._device_offload import dispatch, wrap_output_data
|
|
39
25
|
from .common import KNeighborsDispatchingBase
|
|
40
26
|
|
|
41
|
-
if sklearn_check_version("
|
|
42
|
-
|
|
43
|
-
class NearestNeighbors_(sklearn_NearestNeighbors):
|
|
44
|
-
def __init__(
|
|
45
|
-
self,
|
|
46
|
-
n_neighbors=5,
|
|
47
|
-
radius=1.0,
|
|
48
|
-
algorithm="auto",
|
|
49
|
-
leaf_size=30,
|
|
50
|
-
metric="minkowski",
|
|
51
|
-
p=2,
|
|
52
|
-
metric_params=None,
|
|
53
|
-
n_jobs=None,
|
|
54
|
-
):
|
|
55
|
-
super().__init__(
|
|
56
|
-
n_neighbors=n_neighbors,
|
|
57
|
-
radius=radius,
|
|
58
|
-
algorithm=algorithm,
|
|
59
|
-
leaf_size=leaf_size,
|
|
60
|
-
metric=metric,
|
|
61
|
-
p=p,
|
|
62
|
-
metric_params=metric_params,
|
|
63
|
-
n_jobs=n_jobs,
|
|
64
|
-
)
|
|
65
|
-
|
|
27
|
+
if sklearn_check_version("1.6"):
|
|
28
|
+
from sklearn.utils.validation import validate_data
|
|
66
29
|
else:
|
|
30
|
+
validate_data = _sklearn_NearestNeighbors._validate_data
|
|
67
31
|
|
|
68
|
-
class NearestNeighbors_(sklearn_NearestNeighbors):
|
|
69
|
-
if sklearn_check_version("1.2"):
|
|
70
|
-
_parameter_constraints: dict = {
|
|
71
|
-
**sklearn_NearestNeighbors._parameter_constraints
|
|
72
|
-
}
|
|
73
32
|
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
*,
|
|
78
|
-
n_neighbors=5,
|
|
79
|
-
radius=1.0,
|
|
80
|
-
algorithm="auto",
|
|
81
|
-
leaf_size=30,
|
|
82
|
-
metric="minkowski",
|
|
83
|
-
p=2,
|
|
84
|
-
metric_params=None,
|
|
85
|
-
n_jobs=None,
|
|
86
|
-
):
|
|
87
|
-
super().__init__(
|
|
88
|
-
n_neighbors=n_neighbors,
|
|
89
|
-
radius=radius,
|
|
90
|
-
algorithm=algorithm,
|
|
91
|
-
leaf_size=leaf_size,
|
|
92
|
-
metric=metric,
|
|
93
|
-
p=p,
|
|
94
|
-
metric_params=metric_params,
|
|
95
|
-
n_jobs=n_jobs,
|
|
96
|
-
)
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
@control_n_jobs(decorated_methods=["fit", "kneighbors"])
|
|
100
|
-
class NearestNeighbors(NearestNeighbors_, KNeighborsDispatchingBase):
|
|
33
|
+
@control_n_jobs(decorated_methods=["fit", "kneighbors", "radius_neighbors"])
|
|
34
|
+
class NearestNeighbors(KNeighborsDispatchingBase, _sklearn_NearestNeighbors):
|
|
35
|
+
__doc__ = _sklearn_NearestNeighbors.__doc__
|
|
101
36
|
if sklearn_check_version("1.2"):
|
|
102
|
-
_parameter_constraints: dict = {
|
|
37
|
+
_parameter_constraints: dict = {
|
|
38
|
+
**_sklearn_NearestNeighbors._parameter_constraints
|
|
39
|
+
}
|
|
103
40
|
|
|
104
41
|
@_deprecate_positional_args
|
|
105
42
|
def __init__(
|
|
@@ -125,13 +62,12 @@ class NearestNeighbors(NearestNeighbors_, KNeighborsDispatchingBase):
|
|
|
125
62
|
)
|
|
126
63
|
|
|
127
64
|
def fit(self, X, y=None):
|
|
128
|
-
self._fit_validation(X, y)
|
|
129
65
|
dispatch(
|
|
130
66
|
self,
|
|
131
67
|
"fit",
|
|
132
68
|
{
|
|
133
69
|
"onedal": self.__class__._onedal_fit,
|
|
134
|
-
"sklearn":
|
|
70
|
+
"sklearn": _sklearn_NearestNeighbors.fit,
|
|
135
71
|
},
|
|
136
72
|
X,
|
|
137
73
|
None,
|
|
@@ -148,7 +84,7 @@ class NearestNeighbors(NearestNeighbors_, KNeighborsDispatchingBase):
|
|
|
148
84
|
"kneighbors",
|
|
149
85
|
{
|
|
150
86
|
"onedal": self.__class__._onedal_kneighbors,
|
|
151
|
-
"sklearn":
|
|
87
|
+
"sklearn": _sklearn_NearestNeighbors.kneighbors,
|
|
152
88
|
},
|
|
153
89
|
X,
|
|
154
90
|
n_neighbors=n_neighbors,
|
|
@@ -159,27 +95,41 @@ class NearestNeighbors(NearestNeighbors_, KNeighborsDispatchingBase):
|
|
|
159
95
|
def radius_neighbors(
|
|
160
96
|
self, X=None, radius=None, return_distance=True, sort_results=False
|
|
161
97
|
):
|
|
162
|
-
_onedal_estimator = getattr(self, "_onedal_estimator", None)
|
|
163
|
-
|
|
164
98
|
if (
|
|
165
|
-
_onedal_estimator
|
|
99
|
+
hasattr(self, "_onedal_estimator")
|
|
166
100
|
or getattr(self, "_tree", 0) is None
|
|
167
101
|
and self._fit_method == "kd_tree"
|
|
168
102
|
):
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
103
|
+
_sklearn_NearestNeighbors.fit(self, self._fit_X, getattr(self, "_y", None))
|
|
104
|
+
check_is_fitted(self)
|
|
105
|
+
return dispatch(
|
|
106
|
+
self,
|
|
107
|
+
"radius_neighbors",
|
|
108
|
+
{
|
|
109
|
+
"onedal": None,
|
|
110
|
+
"sklearn": _sklearn_NearestNeighbors.radius_neighbors,
|
|
111
|
+
},
|
|
112
|
+
X,
|
|
113
|
+
radius=radius,
|
|
114
|
+
return_distance=return_distance,
|
|
115
|
+
sort_results=sort_results,
|
|
116
|
+
)
|
|
117
|
+
|
|
118
|
+
def radius_neighbors_graph(
|
|
119
|
+
self, X=None, radius=None, mode="connectivity", sort_results=False
|
|
120
|
+
):
|
|
121
|
+
return dispatch(
|
|
122
|
+
self,
|
|
123
|
+
"radius_neighbors_graph",
|
|
124
|
+
{
|
|
125
|
+
"onedal": None,
|
|
126
|
+
"sklearn": _sklearn_NearestNeighbors.radius_neighbors_graph,
|
|
127
|
+
},
|
|
128
|
+
X,
|
|
129
|
+
radius=radius,
|
|
130
|
+
mode=mode,
|
|
131
|
+
sort_results=sort_results,
|
|
132
|
+
)
|
|
183
133
|
|
|
184
134
|
def _onedal_fit(self, X, y=None, queue=None):
|
|
185
135
|
onedal_params = {
|
|
@@ -219,3 +169,10 @@ class NearestNeighbors(NearestNeighbors_, KNeighborsDispatchingBase):
|
|
|
219
169
|
self._fit_X = self._onedal_estimator._fit_X
|
|
220
170
|
self._fit_method = self._onedal_estimator._fit_method
|
|
221
171
|
self._tree = self._onedal_estimator._tree
|
|
172
|
+
|
|
173
|
+
fit.__doc__ = _sklearn_NearestNeighbors.__doc__
|
|
174
|
+
kneighbors.__doc__ = _sklearn_NearestNeighbors.kneighbors.__doc__
|
|
175
|
+
radius_neighbors.__doc__ = _sklearn_NearestNeighbors.radius_neighbors.__doc__
|
|
176
|
+
radius_neighbors_graph.__doc__ = (
|
|
177
|
+
_sklearn_NearestNeighbors.radius_neighbors_graph.__doc__
|
|
178
|
+
)
|
|
@@ -14,7 +14,6 @@
|
|
|
14
14
|
# limitations under the License.
|
|
15
15
|
# ===============================================================================
|
|
16
16
|
|
|
17
|
-
import numpy as np
|
|
18
17
|
import pytest
|
|
19
18
|
from numpy.testing import assert_allclose
|
|
20
19
|
|
|
@@ -33,7 +32,6 @@ from sklearnex.neighbors import (
|
|
|
33
32
|
|
|
34
33
|
@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
|
|
35
34
|
def test_sklearnex_import_knn_classifier(dataframe, queue):
|
|
36
|
-
|
|
37
35
|
X = _convert_to_dataframe([[0], [1], [2], [3]], sycl_queue=queue, target_df=dataframe)
|
|
38
36
|
y = _convert_to_dataframe([0, 0, 1, 1], sycl_queue=queue, target_df=dataframe)
|
|
39
37
|
neigh = KNeighborsClassifier(n_neighbors=3).fit(X, y)
|
|
@@ -45,36 +43,35 @@ def test_sklearnex_import_knn_classifier(dataframe, queue):
|
|
|
45
43
|
|
|
46
44
|
@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
|
|
47
45
|
def test_sklearnex_import_knn_regression(dataframe, queue):
|
|
48
|
-
|
|
49
46
|
X = _convert_to_dataframe([[0], [1], [2], [3]], sycl_queue=queue, target_df=dataframe)
|
|
50
47
|
y = _convert_to_dataframe([0, 0, 1, 1], sycl_queue=queue, target_df=dataframe)
|
|
51
48
|
neigh = KNeighborsRegressor(n_neighbors=2).fit(X, y)
|
|
52
49
|
y_test = _convert_to_dataframe([[1.5]], sycl_queue=queue, target_df=dataframe)
|
|
53
|
-
pred = _as_numpy(neigh.predict(y_test))
|
|
50
|
+
pred = _as_numpy(neigh.predict(y_test)).squeeze()
|
|
54
51
|
assert "sklearnex" in neigh.__module__
|
|
55
|
-
assert_allclose(pred,
|
|
52
|
+
assert_allclose(pred, 0.5)
|
|
56
53
|
|
|
57
54
|
|
|
55
|
+
@pytest.mark.parametrize("algorithm", ["auto", "brute"])
|
|
58
56
|
@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
|
|
59
57
|
@pytest.mark.parametrize(
|
|
60
58
|
"estimator",
|
|
61
59
|
[LocalOutlierFactor, NearestNeighbors],
|
|
62
60
|
)
|
|
63
|
-
def test_sklearnex_kneighbors(estimator, dataframe, queue):
|
|
64
|
-
|
|
61
|
+
def test_sklearnex_kneighbors(algorithm, estimator, dataframe, queue):
|
|
65
62
|
X = [[0, 0, 2], [1, 0, 0], [0, 0, 1]]
|
|
66
63
|
X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
|
|
67
64
|
test = _convert_to_dataframe([[0, 0, 1.3]], sycl_queue=queue, target_df=dataframe)
|
|
68
|
-
neigh = estimator(n_neighbors=2).fit(X)
|
|
65
|
+
neigh = estimator(n_neighbors=2, algorithm=algorithm).fit(X)
|
|
69
66
|
result = neigh.kneighbors(test, 2, return_distance=False)
|
|
70
67
|
result = _as_numpy(result)
|
|
71
68
|
assert "sklearnex" in neigh.__module__
|
|
72
69
|
assert_allclose(result, [[2, 0]])
|
|
70
|
+
result = neigh.kneighbors()
|
|
73
71
|
|
|
74
72
|
|
|
75
73
|
@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
|
|
76
74
|
def test_sklearnex_import_lof(dataframe, queue):
|
|
77
|
-
|
|
78
75
|
X = [[7, 7, 7], [1, 0, 0], [0, 0, 1], [0, 0, 1]]
|
|
79
76
|
X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
|
|
80
77
|
lof = LocalOutlierFactor(n_neighbors=2)
|
|
@@ -18,11 +18,11 @@ import warnings
|
|
|
18
18
|
|
|
19
19
|
import numpy as np
|
|
20
20
|
from scipy import sparse as sp
|
|
21
|
-
from sklearn.covariance import EmpiricalCovariance as
|
|
21
|
+
from sklearn.covariance import EmpiricalCovariance as _sklearn_EmpiricalCovariance
|
|
22
22
|
from sklearn.utils import check_array
|
|
23
23
|
|
|
24
24
|
from daal4py.sklearn._n_jobs_support import control_n_jobs
|
|
25
|
-
from daal4py.sklearn._utils import sklearn_check_version
|
|
25
|
+
from daal4py.sklearn._utils import daal_check_version, sklearn_check_version
|
|
26
26
|
from onedal.common.hyperparameters import get_hyperparameters
|
|
27
27
|
from onedal.covariance import EmpiricalCovariance as onedal_EmpiricalCovariance
|
|
28
28
|
from sklearnex import config_context
|
|
@@ -31,19 +31,28 @@ from sklearnex.metrics import pairwise_distances
|
|
|
31
31
|
from ..._device_offload import dispatch, wrap_output_data
|
|
32
32
|
from ..._utils import PatchingConditionsChain, register_hyperparameters
|
|
33
33
|
|
|
34
|
+
if sklearn_check_version("1.6"):
|
|
35
|
+
from sklearn.utils.validation import validate_data
|
|
36
|
+
else:
|
|
37
|
+
validate_data = _sklearn_EmpiricalCovariance._validate_data
|
|
38
|
+
|
|
34
39
|
|
|
35
40
|
@register_hyperparameters({"fit": get_hyperparameters("covariance", "compute")})
|
|
36
41
|
@control_n_jobs(decorated_methods=["fit", "mahalanobis"])
|
|
37
|
-
class EmpiricalCovariance(
|
|
38
|
-
__doc__ =
|
|
42
|
+
class EmpiricalCovariance(_sklearn_EmpiricalCovariance):
|
|
43
|
+
__doc__ = _sklearn_EmpiricalCovariance.__doc__
|
|
39
44
|
|
|
40
45
|
if sklearn_check_version("1.2"):
|
|
41
46
|
_parameter_constraints: dict = {
|
|
42
|
-
**
|
|
47
|
+
**_sklearn_EmpiricalCovariance._parameter_constraints,
|
|
43
48
|
}
|
|
44
49
|
|
|
45
50
|
def _save_attributes(self):
|
|
46
51
|
assert hasattr(self, "_onedal_estimator")
|
|
52
|
+
if not daal_check_version((2024, "P", 400)) and self.assume_centered:
|
|
53
|
+
location = self._onedal_estimator.location_[None, :]
|
|
54
|
+
self._onedal_estimator.covariance_ += np.dot(location.T, location)
|
|
55
|
+
self._onedal_estimator.location_ = np.zeros_like(np.squeeze(location))
|
|
47
56
|
self._set_covariance(self._onedal_estimator.covariance_)
|
|
48
57
|
self.location_ = self._onedal_estimator.location_
|
|
49
58
|
|
|
@@ -58,6 +67,7 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
|
|
|
58
67
|
onedal_params = {
|
|
59
68
|
"method": "dense",
|
|
60
69
|
"bias": True,
|
|
70
|
+
"assume_centered": self.assume_centered,
|
|
61
71
|
}
|
|
62
72
|
|
|
63
73
|
self._onedal_estimator = self._onedal_covariance(**onedal_params)
|
|
@@ -73,10 +83,6 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
|
|
|
73
83
|
(X,) = data
|
|
74
84
|
patching_status.and_conditions(
|
|
75
85
|
[
|
|
76
|
-
(
|
|
77
|
-
self.assume_centered == False,
|
|
78
|
-
"assume_centered parameter is not supported on oneDAL side",
|
|
79
|
-
),
|
|
80
86
|
(not sp.issparse(X), "X is sparse. Sparse input is not supported."),
|
|
81
87
|
]
|
|
82
88
|
)
|
|
@@ -90,7 +96,7 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
|
|
|
90
96
|
if sklearn_check_version("1.2"):
|
|
91
97
|
self._validate_params()
|
|
92
98
|
if sklearn_check_version("0.23"):
|
|
93
|
-
X = self
|
|
99
|
+
X = validate_data(self, X, force_all_finite=False)
|
|
94
100
|
else:
|
|
95
101
|
X = check_array(X, force_all_finite=False)
|
|
96
102
|
|
|
@@ -99,7 +105,7 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
|
|
|
99
105
|
"fit",
|
|
100
106
|
{
|
|
101
107
|
"onedal": self.__class__._onedal_fit,
|
|
102
|
-
"sklearn":
|
|
108
|
+
"sklearn": _sklearn_EmpiricalCovariance.fit,
|
|
103
109
|
},
|
|
104
110
|
X,
|
|
105
111
|
)
|
|
@@ -110,7 +116,7 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
|
|
|
110
116
|
@wrap_output_data
|
|
111
117
|
def mahalanobis(self, X):
|
|
112
118
|
if sklearn_check_version("1.0"):
|
|
113
|
-
X = self
|
|
119
|
+
X = validate_data(self, X, reset=False)
|
|
114
120
|
else:
|
|
115
121
|
X = check_array(X)
|
|
116
122
|
|
|
@@ -123,10 +129,10 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
|
|
|
123
129
|
|
|
124
130
|
return np.reshape(dist, (len(X),)) ** 2
|
|
125
131
|
|
|
126
|
-
error_norm = wrap_output_data(
|
|
127
|
-
score = wrap_output_data(
|
|
132
|
+
error_norm = wrap_output_data(_sklearn_EmpiricalCovariance.error_norm)
|
|
133
|
+
score = wrap_output_data(_sklearn_EmpiricalCovariance.score)
|
|
128
134
|
|
|
129
|
-
fit.__doc__ =
|
|
130
|
-
mahalanobis.__doc__ =
|
|
131
|
-
error_norm.__doc__ =
|
|
132
|
-
score.__doc__ =
|
|
135
|
+
fit.__doc__ = _sklearn_EmpiricalCovariance.fit.__doc__
|
|
136
|
+
mahalanobis.__doc__ = _sklearn_EmpiricalCovariance.mahalanobis
|
|
137
|
+
error_norm.__doc__ = _sklearn_EmpiricalCovariance.error_norm.__doc__
|
|
138
|
+
score.__doc__ = _sklearn_EmpiricalCovariance.score.__doc__
|
|
@@ -27,27 +27,40 @@ from onedal.tests.utils._dataframes_support import (
|
|
|
27
27
|
|
|
28
28
|
@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
|
|
29
29
|
@pytest.mark.parametrize("macro_block", [None, 1024])
|
|
30
|
-
|
|
30
|
+
@pytest.mark.parametrize("assume_centered", [True, False])
|
|
31
|
+
def test_sklearnex_import_covariance(dataframe, queue, macro_block, assume_centered):
|
|
31
32
|
from sklearnex.preview.covariance import EmpiricalCovariance
|
|
32
33
|
|
|
33
34
|
X = np.array([[0, 1], [0, 1]])
|
|
35
|
+
|
|
34
36
|
X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
|
|
35
|
-
empcov = EmpiricalCovariance()
|
|
37
|
+
empcov = EmpiricalCovariance(assume_centered=assume_centered)
|
|
36
38
|
if daal_check_version((2024, "P", 0)) and macro_block is not None:
|
|
37
39
|
hparams = empcov.get_hyperparameters("fit")
|
|
38
40
|
hparams.cpu_macro_block = macro_block
|
|
39
41
|
result = empcov.fit(X)
|
|
42
|
+
|
|
40
43
|
expected_covariance = np.array([[0, 0], [0, 0]])
|
|
41
|
-
expected_means = np.array([0,
|
|
44
|
+
expected_means = np.array([0, 0])
|
|
45
|
+
|
|
46
|
+
if assume_centered:
|
|
47
|
+
expected_covariance = np.array([[0, 0], [0, 1]])
|
|
48
|
+
else:
|
|
49
|
+
expected_means = np.array([0, 1])
|
|
42
50
|
|
|
43
51
|
assert_allclose(expected_covariance, result.covariance_)
|
|
44
52
|
assert_allclose(expected_means, result.location_)
|
|
45
53
|
|
|
46
54
|
X = np.array([[1, 2], [3, 6]])
|
|
55
|
+
|
|
47
56
|
X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
|
|
48
57
|
result = empcov.fit(X)
|
|
49
|
-
|
|
50
|
-
|
|
58
|
+
|
|
59
|
+
if assume_centered:
|
|
60
|
+
expected_covariance = np.array([[5, 10], [10, 20]])
|
|
61
|
+
else:
|
|
62
|
+
expected_covariance = np.array([[1, 2], [2, 4]])
|
|
63
|
+
expected_means = np.array([2, 4])
|
|
51
64
|
|
|
52
65
|
assert_allclose(expected_covariance, result.covariance_)
|
|
53
66
|
assert_allclose(expected_means, result.location_)
|
|
@@ -0,0 +1,19 @@
|
|
|
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
|
+
from .incremental_pca import IncrementalPCA
|
|
18
|
+
|
|
19
|
+
__all__ = ["IncrementalPCA"]
|
|
@@ -0,0 +1,233 @@
|
|
|
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
|
+
from sklearn.decomposition import IncrementalPCA as _sklearn_IncrementalPCA
|
|
19
|
+
from sklearn.utils import check_array, gen_batches
|
|
20
|
+
|
|
21
|
+
from daal4py.sklearn._n_jobs_support import control_n_jobs
|
|
22
|
+
from daal4py.sklearn._utils import sklearn_check_version
|
|
23
|
+
from onedal.decomposition import IncrementalPCA as onedal_IncrementalPCA
|
|
24
|
+
|
|
25
|
+
from ..._device_offload import dispatch, wrap_output_data
|
|
26
|
+
from ..._utils import PatchingConditionsChain
|
|
27
|
+
|
|
28
|
+
if sklearn_check_version("1.6"):
|
|
29
|
+
from sklearn.utils.validation import validate_data
|
|
30
|
+
else:
|
|
31
|
+
validate_data = _sklearn_IncrementalPCA._validate_data
|
|
32
|
+
|
|
33
|
+
|
|
34
|
+
@control_n_jobs(
|
|
35
|
+
decorated_methods=["fit", "partial_fit", "transform", "_onedal_finalize_fit"]
|
|
36
|
+
)
|
|
37
|
+
class IncrementalPCA(_sklearn_IncrementalPCA):
|
|
38
|
+
|
|
39
|
+
def __init__(self, n_components=None, *, whiten=False, copy=True, batch_size=None):
|
|
40
|
+
super().__init__(
|
|
41
|
+
n_components=n_components, whiten=whiten, copy=copy, batch_size=batch_size
|
|
42
|
+
)
|
|
43
|
+
self._need_to_finalize = False
|
|
44
|
+
self._need_to_finalize_attrs = {
|
|
45
|
+
"mean_",
|
|
46
|
+
"explained_variance_",
|
|
47
|
+
"explained_variance_ratio_",
|
|
48
|
+
"n_components_",
|
|
49
|
+
"components_",
|
|
50
|
+
"noise_variance_",
|
|
51
|
+
"singular_values_",
|
|
52
|
+
"var_",
|
|
53
|
+
}
|
|
54
|
+
|
|
55
|
+
_onedal_incremental_pca = staticmethod(onedal_IncrementalPCA)
|
|
56
|
+
|
|
57
|
+
def _onedal_transform(self, X, queue=None):
|
|
58
|
+
assert hasattr(self, "_onedal_estimator")
|
|
59
|
+
if self._need_to_finalize:
|
|
60
|
+
self._onedal_finalize_fit()
|
|
61
|
+
X = check_array(X, dtype=[np.float64, np.float32])
|
|
62
|
+
return self._onedal_estimator.predict(X, queue)
|
|
63
|
+
|
|
64
|
+
def _onedal_fit_transform(self, X, queue=None):
|
|
65
|
+
self._onedal_fit(X, queue)
|
|
66
|
+
return self._onedal_transform(X, queue)
|
|
67
|
+
|
|
68
|
+
def _onedal_partial_fit(self, X, check_input=True, queue=None):
|
|
69
|
+
first_pass = not hasattr(self, "_onedal_estimator")
|
|
70
|
+
|
|
71
|
+
if check_input:
|
|
72
|
+
if sklearn_check_version("1.0"):
|
|
73
|
+
X = validate_data(
|
|
74
|
+
self, X, dtype=[np.float64, np.float32], reset=first_pass
|
|
75
|
+
)
|
|
76
|
+
else:
|
|
77
|
+
X = check_array(
|
|
78
|
+
X,
|
|
79
|
+
dtype=[np.float64, np.float32],
|
|
80
|
+
copy=self.copy,
|
|
81
|
+
)
|
|
82
|
+
|
|
83
|
+
n_samples, n_features = X.shape
|
|
84
|
+
|
|
85
|
+
if self.n_components is None:
|
|
86
|
+
if not hasattr(self, "_components_shape"):
|
|
87
|
+
self.n_components_ = min(n_samples, n_features)
|
|
88
|
+
self._components_shape = self.n_components_
|
|
89
|
+
|
|
90
|
+
elif not self.n_components <= n_features:
|
|
91
|
+
raise ValueError(
|
|
92
|
+
"n_components=%r invalid for n_features=%d, need "
|
|
93
|
+
"more rows than columns for IncrementalPCA "
|
|
94
|
+
"processing" % (self.n_components, n_features)
|
|
95
|
+
)
|
|
96
|
+
elif not self.n_components <= n_samples:
|
|
97
|
+
raise ValueError(
|
|
98
|
+
"n_components=%r must be less or equal to "
|
|
99
|
+
"the batch number of samples "
|
|
100
|
+
"%d." % (self.n_components, n_samples)
|
|
101
|
+
)
|
|
102
|
+
else:
|
|
103
|
+
self.n_components_ = self.n_components
|
|
104
|
+
|
|
105
|
+
if not hasattr(self, "n_samples_seen_"):
|
|
106
|
+
self.n_samples_seen_ = n_samples
|
|
107
|
+
else:
|
|
108
|
+
self.n_samples_seen_ += n_samples
|
|
109
|
+
|
|
110
|
+
onedal_params = {"n_components": self.n_components_, "whiten": self.whiten}
|
|
111
|
+
|
|
112
|
+
if not hasattr(self, "_onedal_estimator"):
|
|
113
|
+
self._onedal_estimator = self._onedal_incremental_pca(**onedal_params)
|
|
114
|
+
self._onedal_estimator.partial_fit(X, queue=queue)
|
|
115
|
+
self._need_to_finalize = True
|
|
116
|
+
|
|
117
|
+
def _onedal_finalize_fit(self, queue=None):
|
|
118
|
+
assert hasattr(self, "_onedal_estimator")
|
|
119
|
+
self._onedal_estimator.finalize_fit(queue=queue)
|
|
120
|
+
self._need_to_finalize = False
|
|
121
|
+
|
|
122
|
+
def _onedal_fit(self, X, queue=None):
|
|
123
|
+
if sklearn_check_version("1.2"):
|
|
124
|
+
self._validate_params()
|
|
125
|
+
|
|
126
|
+
if sklearn_check_version("1.0"):
|
|
127
|
+
X = validate_data(self, X, dtype=[np.float64, np.float32], copy=self.copy)
|
|
128
|
+
else:
|
|
129
|
+
X = check_array(
|
|
130
|
+
X,
|
|
131
|
+
dtype=[np.float64, np.float32],
|
|
132
|
+
copy=self.copy,
|
|
133
|
+
)
|
|
134
|
+
|
|
135
|
+
n_samples, n_features = X.shape
|
|
136
|
+
|
|
137
|
+
if self.batch_size is None:
|
|
138
|
+
self.batch_size_ = 5 * n_features
|
|
139
|
+
else:
|
|
140
|
+
self.batch_size_ = self.batch_size
|
|
141
|
+
|
|
142
|
+
self.n_samples_seen_ = 0
|
|
143
|
+
if hasattr(self, "_onedal_estimator"):
|
|
144
|
+
self._onedal_estimator._reset()
|
|
145
|
+
|
|
146
|
+
for batch in gen_batches(n_samples, self.batch_size_):
|
|
147
|
+
X_batch = X[batch]
|
|
148
|
+
self._onedal_partial_fit(X_batch, queue=queue)
|
|
149
|
+
|
|
150
|
+
self._onedal_finalize_fit(queue=queue)
|
|
151
|
+
|
|
152
|
+
return self
|
|
153
|
+
|
|
154
|
+
def _onedal_supported(self, method_name, *data):
|
|
155
|
+
patching_status = PatchingConditionsChain(
|
|
156
|
+
f"sklearn.decomposition.{self.__class__.__name__}.{method_name}"
|
|
157
|
+
)
|
|
158
|
+
return patching_status
|
|
159
|
+
|
|
160
|
+
_onedal_cpu_supported = _onedal_supported
|
|
161
|
+
_onedal_gpu_supported = _onedal_supported
|
|
162
|
+
|
|
163
|
+
def __getattr__(self, attr):
|
|
164
|
+
if attr in self._need_to_finalize_attrs:
|
|
165
|
+
if hasattr(self, "_onedal_estimator"):
|
|
166
|
+
if self._need_to_finalize:
|
|
167
|
+
self._onedal_finalize_fit()
|
|
168
|
+
return getattr(self._onedal_estimator, attr)
|
|
169
|
+
else:
|
|
170
|
+
raise AttributeError(
|
|
171
|
+
f"'{self.__class__.__name__}' object has no attribute '{attr}'"
|
|
172
|
+
)
|
|
173
|
+
if attr in self.__dict__:
|
|
174
|
+
return self.__dict__[attr]
|
|
175
|
+
|
|
176
|
+
raise AttributeError(
|
|
177
|
+
f"'{self.__class__.__name__}' object has no attribute '{attr}'"
|
|
178
|
+
)
|
|
179
|
+
|
|
180
|
+
def partial_fit(self, X, y=None, check_input=True):
|
|
181
|
+
dispatch(
|
|
182
|
+
self,
|
|
183
|
+
"partial_fit",
|
|
184
|
+
{
|
|
185
|
+
"onedal": self.__class__._onedal_partial_fit,
|
|
186
|
+
"sklearn": _sklearn_IncrementalPCA.partial_fit,
|
|
187
|
+
},
|
|
188
|
+
X,
|
|
189
|
+
check_input=check_input,
|
|
190
|
+
)
|
|
191
|
+
return self
|
|
192
|
+
|
|
193
|
+
def fit(self, X, y=None):
|
|
194
|
+
dispatch(
|
|
195
|
+
self,
|
|
196
|
+
"fit",
|
|
197
|
+
{
|
|
198
|
+
"onedal": self.__class__._onedal_fit,
|
|
199
|
+
"sklearn": _sklearn_IncrementalPCA.fit,
|
|
200
|
+
},
|
|
201
|
+
X,
|
|
202
|
+
)
|
|
203
|
+
return self
|
|
204
|
+
|
|
205
|
+
@wrap_output_data
|
|
206
|
+
def transform(self, X):
|
|
207
|
+
return dispatch(
|
|
208
|
+
self,
|
|
209
|
+
"transform",
|
|
210
|
+
{
|
|
211
|
+
"onedal": self.__class__._onedal_transform,
|
|
212
|
+
"sklearn": _sklearn_IncrementalPCA.transform,
|
|
213
|
+
},
|
|
214
|
+
X,
|
|
215
|
+
)
|
|
216
|
+
|
|
217
|
+
@wrap_output_data
|
|
218
|
+
def fit_transform(self, X, y=None, **fit_params):
|
|
219
|
+
return dispatch(
|
|
220
|
+
self,
|
|
221
|
+
"fit_transform",
|
|
222
|
+
{
|
|
223
|
+
"onedal": self.__class__._onedal_fit_transform,
|
|
224
|
+
"sklearn": _sklearn_IncrementalPCA.fit_transform,
|
|
225
|
+
},
|
|
226
|
+
X,
|
|
227
|
+
)
|
|
228
|
+
|
|
229
|
+
__doc__ = _sklearn_IncrementalPCA.__doc__
|
|
230
|
+
fit.__doc__ = _sklearn_IncrementalPCA.fit.__doc__
|
|
231
|
+
fit_transform.__doc__ = _sklearn_IncrementalPCA.fit_transform.__doc__
|
|
232
|
+
transform.__doc__ = _sklearn_IncrementalPCA.transform.__doc__
|
|
233
|
+
partial_fit.__doc__ = _sklearn_IncrementalPCA.partial_fit.__doc__
|