scikit-learn-intelex 2024.7.0__py39-none-win_amd64.whl → 2025.0.1__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.0.1.data/data/Lib/site-packages/daal4py/__init__.py +73 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/__main__.py +58 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/_daal4py.cp39-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/doc/third-party-programs.txt +424 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/mb/__init__.py +19 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/mb/model_builders.py +377 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/mpi_transceiver.cp39-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/__init__.py +40 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/_n_jobs_support.py +242 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/_utils.py +241 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/cluster/__init__.py +20 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/cluster/dbscan.py +165 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/cluster/k_means.py +597 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/cluster/tests/test_dbscan.py +109 -0
- {scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/spmd → scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn}/decomposition/__init__.py +2 -2
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/decomposition/_pca.py +524 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/ensemble/AdaBoostClassifier.py +192 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/ensemble/GBTDAAL.py +318 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/ensemble/__init__.py +27 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/ensemble/_forest.py +1397 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/ensemble/tests/test_decision_forest.py +206 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/__init__.py +29 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_coordinate_descent.py +848 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_linear.py +272 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/_ridge.py +325 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/coordinate_descent.py +17 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/linear.py +17 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/logistic_loss.py +195 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/logistic_path.py +1026 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/ridge.py +17 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_linear.py +196 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/linear_model/tests/test_ridge.py +69 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/manifold/__init__.py +19 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/manifold/_t_sne.py +405 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/metrics/__init__.py +20 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/metrics/_pairwise.py +155 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/metrics/_ranking.py +210 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/model_selection/__init__.py +19 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/model_selection/_split.py +309 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/model_selection/tests/test_split.py +56 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/__init__.py +0 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/dispatcher.py +232 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/_models_info.py +161 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/test_monkeypatch.py +71 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/test_patching.py +87 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/monkeypatch/tests/utils/_launch_algorithms.py +118 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/neighbors/__init__.py +21 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_base.py +503 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_classification.py +139 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_regression.py +74 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/neighbors/_unsupervised.py +55 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/neighbors/tests/test_kneighbors.py +113 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/svm/__init__.py +19 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/svm/svm.py +734 -0
- {scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/spmd/covariance → scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/utils}/__init__.py +5 -3
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/utils/base.py +75 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/utils/tests/test_utils.py +51 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/daal4py/sklearn/utils/validation.py +693 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/__init__.py +83 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/_config.py +53 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/_device_offload.py +229 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/_onedal_py_dpc.cp39-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/_onedal_py_host.cp39-win_amd64.pyd +0 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/basic_statistics/basic_statistics.py +107 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/basic_statistics/incremental_basic_statistics.py +160 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/basic_statistics/tests/test_basic_statistics.py +298 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/basic_statistics/tests/test_incremental_basic_statistics.py +196 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/cluster/__init__.py +27 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/cluster/dbscan.py +110 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/cluster/kmeans.py +560 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/cluster/kmeans_init.py +115 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/cluster/tests/test_dbscan.py +125 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/cluster/tests/test_kmeans.py +88 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/cluster/tests/test_kmeans_init.py +93 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/common/_base.py +38 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/common/_estimator_checks.py +47 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/common/_mixin.py +62 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/common/_policy.py +59 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/common/_spmd_policy.py +30 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/common/hyperparameters.py +116 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/common/tests/test_policy.py +75 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/covariance/__init__.py +20 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/covariance/covariance.py +125 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/covariance/incremental_covariance.py +146 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/covariance/tests/test_covariance.py +50 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/covariance/tests/test_incremental_covariance.py +122 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/datatypes/__init__.py +19 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/datatypes/_data_conversion.py +95 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/datatypes/tests/test_data.py +235 -0
- {scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics → scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/decomposition}/__init__.py +3 -2
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/decomposition/incremental_pca.py +204 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/decomposition/pca.py +186 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/decomposition/tests/test_incremental_pca.py +198 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/ensemble/__init__.py +29 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/ensemble/forest.py +720 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/ensemble/tests/test_random_forest.py +97 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/linear_model/__init__.py +27 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/linear_model/incremental_linear_model.py +258 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/linear_model/linear_model.py +329 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/linear_model/logistic_regression.py +249 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/linear_model/tests/test_incremental_linear_regression.py +168 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/linear_model/tests/test_incremental_ridge_regression.py +107 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/linear_model/tests/test_linear_regression.py +149 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/linear_model/tests/test_logistic_regression.py +95 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/linear_model/tests/test_ridge.py +95 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/neighbors/__init__.py +19 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/neighbors/neighbors.py +778 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/neighbors/tests/test_knn_classification.py +49 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/primitives/__init__.py +27 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/primitives/get_tree.py +25 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/primitives/kernel_functions.py +153 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/primitives/tests/test_kernel_functions.py +159 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/svm/__init__.py +19 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/svm/svm.py +556 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/svm/tests/test_csr_svm.py +351 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/svm/tests/test_nusvc.py +204 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/svm/tests/test_nusvr.py +210 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/svm/tests/test_svc.py +168 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/svm/tests/test_svr.py +243 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/tests/test_common.py +41 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/tests/utils/_dataframes_support.py +168 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/tests/utils/_device_selection.py +107 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/utils/__init__.py +49 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/utils/_array_api.py +91 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal/utils/validation.py +432 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/_device_offload.py +36 -13
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/basic_statistics/__init__.py +20 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/basic_statistics/basic_statistics.py +30 -8
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/basic_statistics/incremental_basic_statistics.py +49 -16
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/cluster/__init__.py +1 -1
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/cluster/k_means.py +383 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/cluster/tests/test_kmeans.py +153 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/covariance/incremental_covariance.py +28 -10
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/covariance/tests/test_incremental_covariance.py +11 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/decomposition/pca.py +1 -1
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/dispatcher.py +19 -9
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/ensemble/_forest.py +1 -1
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/__init__.py +2 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/coordinate_descent.py +7 -7
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/incremental_linear.py +45 -26
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/linear_model/incremental_ridge.py +418 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/ridge.py +4 -4
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_linear.py +13 -10
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_ridge.py +153 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/tests/test_logreg.py +5 -4
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/manifold/t_sne.py +3 -3
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/metrics/pairwise.py +2 -2
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/metrics/ranking.py +2 -2
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/model_selection/split.py +4 -2
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/neighbors/_lof.py +2 -1
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/neighbors/common.py +1 -1
- {scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/preview/cluster → scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/preview}/__init__.py +1 -3
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/preview/decomposition/incremental_pca.py +8 -8
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/preview/decomposition/tests/test_incremental_pca.py +2 -2
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/preview/linear_model/ridge.py +4 -4
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/__init__.py +20 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/incremental_basic_statistics.py +30 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_incremental_basic_statistics_spmd.py +307 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/covariance/__init__.py +20 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/covariance/incremental_covariance.py +37 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_incremental_covariance_spmd.py +184 -0
- {scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/preview → scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/decomposition}/__init__.py +4 -1
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/decomposition/incremental_pca.py +30 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_incremental_pca_spmd.py +269 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/__init__.py +2 -1
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/linear_model/incremental_linear_model.py +35 -0
- scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_incremental_linear_spmd.py +329 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py +4 -1
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/svm/nusvc.py +1 -1
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/svm/svc.py +1 -1
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/_utils_spmd.py +18 -5
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/test_memory_usage.py +2 -1
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/test_patching.py +0 -1
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/test_run_to_run_stability.py +12 -11
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/utils/__init__.py +1 -2
- scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/utils/_namespace.py → scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/utils/_array_api.py +5 -20
- {scikit_learn_intelex-2024.7.0.dist-info → scikit_learn_intelex-2025.0.1.dist-info}/METADATA +3 -2
- scikit_learn_intelex-2025.0.1.dist-info/RECORD +255 -0
- scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/cluster/k_means.py +0 -25
- scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/cluster/tests/test_kmeans.py +0 -42
- scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/preview/cluster/_common.py +0 -84
- scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex/preview/cluster/k_means.py +0 -303
- scikit_learn_intelex-2024.7.0.dist-info/RECORD +0 -122
- {scikit_learn_intelex-2024.7.0.data/data/Lib/site-packages/sklearnex → scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/onedal}/basic_statistics/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/__main__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/_config.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/_utils.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/basic_statistics/tests/test_basic_statistics.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/basic_statistics/tests/test_incremental_basic_statistics.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/cluster/dbscan.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/cluster/tests/test_dbscan.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/conftest.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/covariance/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/decomposition/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/decomposition/tests/test_pca.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/doc/third-party-programs.txt +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/ensemble/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/ensemble/tests/test_forest.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/glob/__main__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/glob/dispatcher.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/linear.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/logistic_path.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/logistic_regression.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/linear_model/tests/test_linear.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/manifold/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/manifold/tests/test_tsne.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/metrics/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/metrics/tests/test_metrics.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/model_selection/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/model_selection/tests/test_model_selection.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/neighbors/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/neighbors/knn_classification.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/neighbors/knn_regression.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/neighbors/knn_unsupervised.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/neighbors/tests/test_neighbors.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/preview/covariance/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/preview/covariance/covariance.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/preview/covariance/tests/test_covariance.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/preview/decomposition/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/preview/linear_model/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/preview/linear_model/tests/test_ridge.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/basic_statistics/basic_statistics.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_basic_statistics_spmd.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/cluster/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/cluster/dbscan.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/cluster/kmeans.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_dbscan_spmd.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_kmeans_spmd.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/covariance/covariance.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_covariance_spmd.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/decomposition/pca.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_pca_spmd.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/ensemble/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/ensemble/forest.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/ensemble/tests/test_forest_spmd.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/linear_model.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/logistic_regression.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_linear_regression_spmd.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/neighbors/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/neighbors/neighbors.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/spmd/neighbors/tests/test_neighbors_spmd.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/svm/__init__.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/svm/_common.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/svm/nusvr.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/svm/svr.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/svm/tests/test_svm.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/_utils.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/test_common.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/test_config.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/test_monkeypatch.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/test_n_jobs_support.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/tests/test_parallel.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/utils/parallel.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/utils/tests/test_finite.py +0 -0
- {scikit_learn_intelex-2024.7.0.data → scikit_learn_intelex-2025.0.1.data}/data/Lib/site-packages/sklearnex/utils/validation.py +0 -0
- {scikit_learn_intelex-2024.7.0.dist-info → scikit_learn_intelex-2025.0.1.dist-info}/LICENSE.txt +0 -0
- {scikit_learn_intelex-2024.7.0.dist-info → scikit_learn_intelex-2025.0.1.dist-info}/WHEEL +0 -0
- {scikit_learn_intelex-2024.7.0.dist-info → scikit_learn_intelex-2025.0.1.dist-info}/top_level.txt +0 -0
|
@@ -15,8 +15,8 @@
|
|
|
15
15
|
# ===============================================================================
|
|
16
16
|
|
|
17
17
|
from daal4py.sklearn.metrics import pairwise_distances
|
|
18
|
-
from onedal._device_offload import
|
|
18
|
+
from onedal._device_offload import support_input_format
|
|
19
19
|
|
|
20
|
-
pairwise_distances =
|
|
20
|
+
pairwise_distances = support_input_format(freefunc=True, queue_param=False)(
|
|
21
21
|
pairwise_distances
|
|
22
22
|
)
|
|
@@ -15,6 +15,6 @@
|
|
|
15
15
|
# ===============================================================================
|
|
16
16
|
|
|
17
17
|
from daal4py.sklearn.metrics import roc_auc_score
|
|
18
|
-
from onedal._device_offload import
|
|
18
|
+
from onedal._device_offload import support_input_format
|
|
19
19
|
|
|
20
|
-
roc_auc_score =
|
|
20
|
+
roc_auc_score = support_input_format(freefunc=True, queue_param=False)(roc_auc_score)
|
|
@@ -15,6 +15,8 @@
|
|
|
15
15
|
# ===============================================================================
|
|
16
16
|
|
|
17
17
|
from daal4py.sklearn.model_selection import train_test_split
|
|
18
|
-
from onedal._device_offload import
|
|
18
|
+
from onedal._device_offload import support_input_format
|
|
19
19
|
|
|
20
|
-
train_test_split =
|
|
20
|
+
train_test_split = support_input_format(freefunc=True, queue_param=False)(
|
|
21
|
+
train_test_split
|
|
22
|
+
)
|
|
@@ -26,7 +26,8 @@ from daal4py.sklearn._utils import sklearn_check_version
|
|
|
26
26
|
from sklearnex._device_offload import dispatch, wrap_output_data
|
|
27
27
|
from sklearnex.neighbors.common import KNeighborsDispatchingBase
|
|
28
28
|
from sklearnex.neighbors.knn_unsupervised import NearestNeighbors
|
|
29
|
-
|
|
29
|
+
|
|
30
|
+
from ..utils._array_api import get_namespace
|
|
30
31
|
|
|
31
32
|
|
|
32
33
|
@control_n_jobs(decorated_methods=["fit", "_kneighbors"])
|
|
@@ -28,7 +28,7 @@ from daal4py.sklearn._utils import sklearn_check_version
|
|
|
28
28
|
from onedal.utils import _check_array, _num_features, _num_samples
|
|
29
29
|
|
|
30
30
|
from .._utils import PatchingConditionsChain
|
|
31
|
-
from ..utils import get_namespace
|
|
31
|
+
from ..utils._array_api import get_namespace
|
|
32
32
|
|
|
33
33
|
|
|
34
34
|
class KNeighborsDispatchingBase:
|
|
@@ -61,7 +61,7 @@ class IncrementalPCA(sklearn_IncrementalPCA):
|
|
|
61
61
|
return self._onedal_transform(X, queue)
|
|
62
62
|
|
|
63
63
|
def _onedal_partial_fit(self, X, check_input=True, queue=None):
|
|
64
|
-
first_pass = not hasattr(self, "
|
|
64
|
+
first_pass = not hasattr(self, "_onedal_estimator")
|
|
65
65
|
|
|
66
66
|
if check_input:
|
|
67
67
|
if sklearn_check_version("1.0"):
|
|
@@ -78,10 +78,10 @@ class IncrementalPCA(sklearn_IncrementalPCA):
|
|
|
78
78
|
n_samples, n_features = X.shape
|
|
79
79
|
|
|
80
80
|
if self.n_components is None:
|
|
81
|
-
if not hasattr(self, "
|
|
81
|
+
if not hasattr(self, "_components_shape"):
|
|
82
82
|
self.n_components_ = min(n_samples, n_features)
|
|
83
|
-
|
|
84
|
-
|
|
83
|
+
self._components_shape = self.n_components_
|
|
84
|
+
|
|
85
85
|
elif not self.n_components <= n_features:
|
|
86
86
|
raise ValueError(
|
|
87
87
|
"n_components=%r invalid for n_features=%d, need "
|
|
@@ -106,12 +106,12 @@ class IncrementalPCA(sklearn_IncrementalPCA):
|
|
|
106
106
|
|
|
107
107
|
if not hasattr(self, "_onedal_estimator"):
|
|
108
108
|
self._onedal_estimator = self._onedal_incremental_pca(**onedal_params)
|
|
109
|
-
self._onedal_estimator.partial_fit(X, queue)
|
|
109
|
+
self._onedal_estimator.partial_fit(X, queue=queue)
|
|
110
110
|
self._need_to_finalize = True
|
|
111
111
|
|
|
112
|
-
def _onedal_finalize_fit(self):
|
|
112
|
+
def _onedal_finalize_fit(self, queue=None):
|
|
113
113
|
assert hasattr(self, "_onedal_estimator")
|
|
114
|
-
self._onedal_estimator.finalize_fit()
|
|
114
|
+
self._onedal_estimator.finalize_fit(queue=queue)
|
|
115
115
|
self._need_to_finalize = False
|
|
116
116
|
|
|
117
117
|
def _onedal_fit(self, X, queue=None):
|
|
@@ -142,7 +142,7 @@ class IncrementalPCA(sklearn_IncrementalPCA):
|
|
|
142
142
|
X_batch = X[batch]
|
|
143
143
|
self._onedal_partial_fit(X_batch, queue=queue)
|
|
144
144
|
|
|
145
|
-
self._onedal_finalize_fit()
|
|
145
|
+
self._onedal_finalize_fit(queue=queue)
|
|
146
146
|
|
|
147
147
|
return self
|
|
148
148
|
|
|
@@ -74,7 +74,7 @@ def check_pca_on_gold_data(incpca, dtype, whiten, transformed_data):
|
|
|
74
74
|
)
|
|
75
75
|
|
|
76
76
|
tol = 1e-7
|
|
77
|
-
if
|
|
77
|
+
if dtype == np.float32:
|
|
78
78
|
tol = 7e-6 if whiten else 1e-6
|
|
79
79
|
|
|
80
80
|
assert incpca.n_samples_seen_ == expected_n_samples_seen_
|
|
@@ -112,7 +112,7 @@ def check_pca_on_gold_data(incpca, dtype, whiten, transformed_data):
|
|
|
112
112
|
|
|
113
113
|
|
|
114
114
|
def check_pca(incpca, dtype, whiten, data, transformed_data):
|
|
115
|
-
tol = 3e-3 if
|
|
115
|
+
tol = 3e-3 if dtype == np.float32 else 2e-6
|
|
116
116
|
|
|
117
117
|
n_components = incpca.n_components_
|
|
118
118
|
|
|
@@ -408,11 +408,11 @@ if daal_check_version((2024, "P", 600)):
|
|
|
408
408
|
|
|
409
409
|
else:
|
|
410
410
|
from daal4py.sklearn.linear_model._ridge import Ridge
|
|
411
|
-
from onedal._device_offload import
|
|
411
|
+
from onedal._device_offload import support_input_format
|
|
412
412
|
|
|
413
|
-
Ridge.fit =
|
|
414
|
-
Ridge.predict =
|
|
415
|
-
Ridge.score =
|
|
413
|
+
Ridge.fit = support_input_format(queue_param=False)(Ridge.fit)
|
|
414
|
+
Ridge.predict = support_input_format(queue_param=False)(Ridge.predict)
|
|
415
|
+
Ridge.score = support_input_format(queue_param=False)(Ridge.score)
|
|
416
416
|
|
|
417
417
|
logging.warning(
|
|
418
418
|
"Preview Ridge requires oneDAL version >= 2024.6 but it was not found"
|
|
@@ -0,0 +1,20 @@
|
|
|
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
|
+
from .basic_statistics import BasicStatistics
|
|
18
|
+
from .incremental_basic_statistics import IncrementalBasicStatistics
|
|
19
|
+
|
|
20
|
+
__all__ = ["BasicStatistics", "IncrementalBasicStatistics"]
|
|
@@ -0,0 +1,30 @@
|
|
|
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
|
+
|
|
18
|
+
from onedal.spmd.basic_statistics import (
|
|
19
|
+
IncrementalBasicStatistics as onedalSPMD_IncrementalBasicStatistics,
|
|
20
|
+
)
|
|
21
|
+
|
|
22
|
+
from ...basic_statistics import (
|
|
23
|
+
IncrementalBasicStatistics as base_IncrementalBasicStatistics,
|
|
24
|
+
)
|
|
25
|
+
|
|
26
|
+
|
|
27
|
+
class IncrementalBasicStatistics(base_IncrementalBasicStatistics):
|
|
28
|
+
_onedal_incremental_basic_statistics = staticmethod(
|
|
29
|
+
onedalSPMD_IncrementalBasicStatistics
|
|
30
|
+
)
|
|
@@ -0,0 +1,307 @@
|
|
|
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
|
+
import pytest
|
|
19
|
+
from numpy.testing import assert_allclose
|
|
20
|
+
|
|
21
|
+
from onedal.basic_statistics.tests.test_basic_statistics import options_and_tests
|
|
22
|
+
from onedal.tests.utils._dataframes_support import (
|
|
23
|
+
_convert_to_dataframe,
|
|
24
|
+
get_dataframes_and_queues,
|
|
25
|
+
)
|
|
26
|
+
from sklearnex.tests._utils_spmd import (
|
|
27
|
+
_generate_statistic_data,
|
|
28
|
+
_get_local_tensor,
|
|
29
|
+
_mpi_libs_and_gpu_available,
|
|
30
|
+
)
|
|
31
|
+
|
|
32
|
+
|
|
33
|
+
@pytest.mark.skipif(
|
|
34
|
+
not _mpi_libs_and_gpu_available,
|
|
35
|
+
reason="GPU device and MPI libs required for test",
|
|
36
|
+
)
|
|
37
|
+
@pytest.mark.parametrize(
|
|
38
|
+
"dataframe,queue",
|
|
39
|
+
get_dataframes_and_queues(dataframe_filter_="dpnp,dpctl", device_filter_="gpu"),
|
|
40
|
+
)
|
|
41
|
+
@pytest.mark.parametrize("weighted", [True, False])
|
|
42
|
+
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
|
|
43
|
+
@pytest.mark.mpi
|
|
44
|
+
def test_incremental_basic_statistics_fit_spmd_gold(dataframe, queue, weighted, dtype):
|
|
45
|
+
# Import spmd and batch algo
|
|
46
|
+
from sklearnex.basic_statistics import IncrementalBasicStatistics
|
|
47
|
+
from sklearnex.spmd.basic_statistics import (
|
|
48
|
+
IncrementalBasicStatistics as IncrementalBasicStatistics_SPMD,
|
|
49
|
+
)
|
|
50
|
+
|
|
51
|
+
# Create gold data and process into dpt
|
|
52
|
+
data = np.array(
|
|
53
|
+
[
|
|
54
|
+
[0.0, 0.0, 0.0],
|
|
55
|
+
[0.0, 1.0, 2.0],
|
|
56
|
+
[0.0, 2.0, 4.0],
|
|
57
|
+
[0.0, 3.0, 8.0],
|
|
58
|
+
[0.0, 4.0, 16.0],
|
|
59
|
+
[0.0, 5.0, 32.0],
|
|
60
|
+
[0.0, 6.0, 64.0],
|
|
61
|
+
[0.0, 7.0, 128.0],
|
|
62
|
+
],
|
|
63
|
+
dtype=dtype,
|
|
64
|
+
)
|
|
65
|
+
dpt_data = _convert_to_dataframe(data, sycl_queue=queue, target_df=dataframe)
|
|
66
|
+
|
|
67
|
+
local_dpt_data = _convert_to_dataframe(
|
|
68
|
+
_get_local_tensor(data), sycl_queue=queue, target_df=dataframe
|
|
69
|
+
)
|
|
70
|
+
|
|
71
|
+
if weighted:
|
|
72
|
+
# Create weights array containing the weight for each sample in the data
|
|
73
|
+
weights = np.array([0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], dtype=dtype)
|
|
74
|
+
dpt_weights = _convert_to_dataframe(
|
|
75
|
+
weights, sycl_queue=queue, target_df=dataframe
|
|
76
|
+
)
|
|
77
|
+
local_dpt_weights = _convert_to_dataframe(
|
|
78
|
+
_get_local_tensor(weights), sycl_queue=queue, target_df=dataframe
|
|
79
|
+
)
|
|
80
|
+
|
|
81
|
+
# ensure results of batch algo match spmd
|
|
82
|
+
|
|
83
|
+
incbs_spmd = IncrementalBasicStatistics_SPMD().fit(
|
|
84
|
+
local_dpt_data, sample_weight=local_dpt_weights if weighted else None
|
|
85
|
+
)
|
|
86
|
+
incbs = IncrementalBasicStatistics().fit(
|
|
87
|
+
dpt_data, sample_weight=dpt_weights if weighted else None
|
|
88
|
+
)
|
|
89
|
+
|
|
90
|
+
for option, _, _ in options_and_tests:
|
|
91
|
+
assert_allclose(
|
|
92
|
+
getattr(incbs_spmd, option),
|
|
93
|
+
getattr(incbs, option),
|
|
94
|
+
err_msg=f"Result for {option} is incorrect",
|
|
95
|
+
)
|
|
96
|
+
|
|
97
|
+
|
|
98
|
+
@pytest.mark.skipif(
|
|
99
|
+
not _mpi_libs_and_gpu_available,
|
|
100
|
+
reason="GPU device and MPI libs required for test",
|
|
101
|
+
)
|
|
102
|
+
@pytest.mark.parametrize(
|
|
103
|
+
"dataframe,queue",
|
|
104
|
+
get_dataframes_and_queues(dataframe_filter_="dpnp,dpctl", device_filter_="gpu"),
|
|
105
|
+
)
|
|
106
|
+
@pytest.mark.parametrize("num_blocks", [1, 2])
|
|
107
|
+
@pytest.mark.parametrize("weighted", [True, False])
|
|
108
|
+
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
|
|
109
|
+
@pytest.mark.mpi
|
|
110
|
+
def test_incremental_basic_statistics_partial_fit_spmd_gold(
|
|
111
|
+
dataframe, queue, num_blocks, weighted, dtype
|
|
112
|
+
):
|
|
113
|
+
# Import spmd and batch algo
|
|
114
|
+
from sklearnex.basic_statistics import IncrementalBasicStatistics
|
|
115
|
+
from sklearnex.spmd.basic_statistics import (
|
|
116
|
+
IncrementalBasicStatistics as IncrementalBasicStatistics_SPMD,
|
|
117
|
+
)
|
|
118
|
+
|
|
119
|
+
# Create gold data and process into dpt
|
|
120
|
+
data = np.array(
|
|
121
|
+
[
|
|
122
|
+
[0.0, 0.0, 0.0],
|
|
123
|
+
[0.0, 1.0, 2.0],
|
|
124
|
+
[0.0, 2.0, 4.0],
|
|
125
|
+
[0.0, 3.0, 8.0],
|
|
126
|
+
[0.0, 4.0, 16.0],
|
|
127
|
+
[0.0, 5.0, 32.0],
|
|
128
|
+
[0.0, 6.0, 64.0],
|
|
129
|
+
[0.0, 7.0, 128.0],
|
|
130
|
+
],
|
|
131
|
+
dtype=dtype,
|
|
132
|
+
)
|
|
133
|
+
dpt_data = _convert_to_dataframe(data, sycl_queue=queue, target_df=dataframe)
|
|
134
|
+
local_data = _get_local_tensor(data)
|
|
135
|
+
split_local_data = np.array_split(local_data, num_blocks)
|
|
136
|
+
|
|
137
|
+
if weighted:
|
|
138
|
+
# Create weights array containing the weight for each sample in the data
|
|
139
|
+
weights = np.array([0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], dtype=dtype)
|
|
140
|
+
dpt_weights = _convert_to_dataframe(
|
|
141
|
+
weights, sycl_queue=queue, target_df=dataframe
|
|
142
|
+
)
|
|
143
|
+
local_weights = _get_local_tensor(weights)
|
|
144
|
+
split_local_weights = np.array_split(local_weights, num_blocks)
|
|
145
|
+
|
|
146
|
+
incbs_spmd = IncrementalBasicStatistics_SPMD()
|
|
147
|
+
incbs = IncrementalBasicStatistics()
|
|
148
|
+
|
|
149
|
+
for i in range(num_blocks):
|
|
150
|
+
local_dpt_data = _convert_to_dataframe(
|
|
151
|
+
split_local_data[i], sycl_queue=queue, target_df=dataframe
|
|
152
|
+
)
|
|
153
|
+
if weighted:
|
|
154
|
+
local_dpt_weights = _convert_to_dataframe(
|
|
155
|
+
split_local_weights[i], sycl_queue=queue, target_df=dataframe
|
|
156
|
+
)
|
|
157
|
+
incbs_spmd.partial_fit(
|
|
158
|
+
local_dpt_data, sample_weight=local_dpt_weights if weighted else None
|
|
159
|
+
)
|
|
160
|
+
|
|
161
|
+
incbs.fit(dpt_data, sample_weight=dpt_weights if weighted else None)
|
|
162
|
+
|
|
163
|
+
for option, _, _ in options_and_tests:
|
|
164
|
+
assert_allclose(
|
|
165
|
+
getattr(incbs_spmd, option),
|
|
166
|
+
getattr(incbs, option),
|
|
167
|
+
err_msg=f"Result for {option} is incorrect",
|
|
168
|
+
)
|
|
169
|
+
|
|
170
|
+
|
|
171
|
+
@pytest.mark.skipif(
|
|
172
|
+
not _mpi_libs_and_gpu_available,
|
|
173
|
+
reason="GPU device and MPI libs required for test",
|
|
174
|
+
)
|
|
175
|
+
@pytest.mark.parametrize(
|
|
176
|
+
"dataframe,queue",
|
|
177
|
+
get_dataframes_and_queues(dataframe_filter_="dpnp,dpctl", device_filter_="gpu"),
|
|
178
|
+
)
|
|
179
|
+
@pytest.mark.parametrize("num_blocks", [1, 2])
|
|
180
|
+
@pytest.mark.parametrize("weighted", [True, False])
|
|
181
|
+
@pytest.mark.parametrize("result_option", options_and_tests)
|
|
182
|
+
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
|
|
183
|
+
@pytest.mark.mpi
|
|
184
|
+
def test_incremental_basic_statistics_single_option_partial_fit_spmd_gold(
|
|
185
|
+
dataframe, queue, num_blocks, weighted, result_option, dtype
|
|
186
|
+
):
|
|
187
|
+
# Import spmd and batch algo
|
|
188
|
+
from sklearnex.basic_statistics import IncrementalBasicStatistics
|
|
189
|
+
from sklearnex.spmd.basic_statistics import (
|
|
190
|
+
IncrementalBasicStatistics as IncrementalBasicStatistics_SPMD,
|
|
191
|
+
)
|
|
192
|
+
|
|
193
|
+
# Create gold data and process into dpt
|
|
194
|
+
data = np.array(
|
|
195
|
+
[
|
|
196
|
+
[0.0, 0.0, 0.0],
|
|
197
|
+
[0.0, 1.0, 2.0],
|
|
198
|
+
[0.0, 2.0, 4.0],
|
|
199
|
+
[0.0, 3.0, 8.0],
|
|
200
|
+
[0.0, 4.0, 16.0],
|
|
201
|
+
[0.0, 5.0, 32.0],
|
|
202
|
+
[0.0, 6.0, 64.0],
|
|
203
|
+
[0.0, 7.0, 128.0],
|
|
204
|
+
],
|
|
205
|
+
dtype=dtype,
|
|
206
|
+
)
|
|
207
|
+
dpt_data = _convert_to_dataframe(data, sycl_queue=queue, target_df=dataframe)
|
|
208
|
+
local_data = _get_local_tensor(data)
|
|
209
|
+
split_local_data = np.array_split(local_data, num_blocks)
|
|
210
|
+
|
|
211
|
+
if weighted:
|
|
212
|
+
# Create weights array containing the weight for each sample in the data
|
|
213
|
+
weights = np.array([0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], dtype=dtype)
|
|
214
|
+
dpt_weights = _convert_to_dataframe(
|
|
215
|
+
weights, sycl_queue=queue, target_df=dataframe
|
|
216
|
+
)
|
|
217
|
+
local_weights = _get_local_tensor(weights)
|
|
218
|
+
split_local_weights = np.array_split(local_weights, num_blocks)
|
|
219
|
+
|
|
220
|
+
option, _, _ = result_option
|
|
221
|
+
incbs_spmd = IncrementalBasicStatistics_SPMD(result_options=option)
|
|
222
|
+
incbs = IncrementalBasicStatistics(result_options=option)
|
|
223
|
+
|
|
224
|
+
for i in range(num_blocks):
|
|
225
|
+
local_dpt_data = _convert_to_dataframe(
|
|
226
|
+
split_local_data[i], sycl_queue=queue, target_df=dataframe
|
|
227
|
+
)
|
|
228
|
+
if weighted:
|
|
229
|
+
local_dpt_weights = _convert_to_dataframe(
|
|
230
|
+
split_local_weights[i], sycl_queue=queue, target_df=dataframe
|
|
231
|
+
)
|
|
232
|
+
incbs_spmd.partial_fit(
|
|
233
|
+
local_dpt_data, sample_weight=local_dpt_weights if weighted else None
|
|
234
|
+
)
|
|
235
|
+
|
|
236
|
+
incbs.fit(dpt_data, sample_weight=dpt_weights if weighted else None)
|
|
237
|
+
|
|
238
|
+
assert_allclose(getattr(incbs_spmd, option), getattr(incbs, option))
|
|
239
|
+
|
|
240
|
+
|
|
241
|
+
@pytest.mark.skipif(
|
|
242
|
+
not _mpi_libs_and_gpu_available,
|
|
243
|
+
reason="GPU device and MPI libs required for test",
|
|
244
|
+
)
|
|
245
|
+
@pytest.mark.parametrize(
|
|
246
|
+
"dataframe,queue",
|
|
247
|
+
get_dataframes_and_queues(dataframe_filter_="dpnp,dpctl", device_filter_="gpu"),
|
|
248
|
+
)
|
|
249
|
+
@pytest.mark.parametrize("num_blocks", [1, 2])
|
|
250
|
+
@pytest.mark.parametrize("weighted", [True, False])
|
|
251
|
+
@pytest.mark.parametrize("n_samples", [100, 10000])
|
|
252
|
+
@pytest.mark.parametrize("n_features", [10, 100])
|
|
253
|
+
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
|
|
254
|
+
@pytest.mark.mpi
|
|
255
|
+
def test_incremental_basic_statistics_partial_fit_spmd_synthetic(
|
|
256
|
+
dataframe, queue, num_blocks, weighted, n_samples, n_features, dtype
|
|
257
|
+
):
|
|
258
|
+
# Import spmd and batch algo
|
|
259
|
+
from sklearnex.basic_statistics import IncrementalBasicStatistics
|
|
260
|
+
from sklearnex.spmd.basic_statistics import (
|
|
261
|
+
IncrementalBasicStatistics as IncrementalBasicStatistics_SPMD,
|
|
262
|
+
)
|
|
263
|
+
|
|
264
|
+
tol = 2e-3 if dtype == np.float32 else 1e-7
|
|
265
|
+
|
|
266
|
+
# Create gold data and process into dpt
|
|
267
|
+
data = _generate_statistic_data(n_samples, n_features, dtype=dtype)
|
|
268
|
+
local_data = _get_local_tensor(data)
|
|
269
|
+
split_local_data = np.array_split(local_data, num_blocks)
|
|
270
|
+
split_data = np.array_split(data, num_blocks)
|
|
271
|
+
|
|
272
|
+
if weighted:
|
|
273
|
+
# Create weights array containing the weight for each sample in the data
|
|
274
|
+
weights = _generate_statistic_data(n_samples, dtype=dtype)
|
|
275
|
+
local_weights = _get_local_tensor(weights)
|
|
276
|
+
split_local_weights = np.array_split(local_weights, num_blocks)
|
|
277
|
+
split_weights = np.array_split(weights, num_blocks)
|
|
278
|
+
|
|
279
|
+
incbs_spmd = IncrementalBasicStatistics_SPMD()
|
|
280
|
+
incbs = IncrementalBasicStatistics()
|
|
281
|
+
|
|
282
|
+
for i in range(num_blocks):
|
|
283
|
+
local_dpt_data = _convert_to_dataframe(
|
|
284
|
+
split_local_data[i], sycl_queue=queue, target_df=dataframe
|
|
285
|
+
)
|
|
286
|
+
dpt_data = _convert_to_dataframe(
|
|
287
|
+
split_data[i], sycl_queue=queue, target_df=dataframe
|
|
288
|
+
)
|
|
289
|
+
if weighted:
|
|
290
|
+
local_dpt_weights = _convert_to_dataframe(
|
|
291
|
+
split_local_weights[i], sycl_queue=queue, target_df=dataframe
|
|
292
|
+
)
|
|
293
|
+
dpt_weights = _convert_to_dataframe(
|
|
294
|
+
split_weights[i], sycl_queue=queue, target_df=dataframe
|
|
295
|
+
)
|
|
296
|
+
incbs_spmd.partial_fit(
|
|
297
|
+
local_dpt_data, sample_weight=local_dpt_weights if weighted else None
|
|
298
|
+
)
|
|
299
|
+
incbs.partial_fit(dpt_data, sample_weight=dpt_weights if weighted else None)
|
|
300
|
+
|
|
301
|
+
for option, _, _ in options_and_tests:
|
|
302
|
+
assert_allclose(
|
|
303
|
+
getattr(incbs_spmd, option),
|
|
304
|
+
getattr(incbs, option),
|
|
305
|
+
atol=tol,
|
|
306
|
+
err_msg=f"Result for {option} is incorrect",
|
|
307
|
+
)
|
scikit_learn_intelex-2025.0.1.data/data/Lib/site-packages/sklearnex/spmd/covariance/__init__.py
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
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 .covariance import EmpiricalCovariance
|
|
18
|
+
from .incremental_covariance import IncrementalEmpiricalCovariance
|
|
19
|
+
|
|
20
|
+
__all__ = ["EmpiricalCovariance", "IncrementalEmpiricalCovariance"]
|
|
@@ -0,0 +1,37 @@
|
|
|
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 onedal.spmd.covariance import (
|
|
18
|
+
IncrementalEmpiricalCovariance as onedalSPMD_IncrementalEmpiricalCovariance,
|
|
19
|
+
)
|
|
20
|
+
|
|
21
|
+
from ...covariance import (
|
|
22
|
+
IncrementalEmpiricalCovariance as base_IncrementalEmpiricalCovariance,
|
|
23
|
+
)
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
class IncrementalEmpiricalCovariance(base_IncrementalEmpiricalCovariance):
|
|
27
|
+
"""
|
|
28
|
+
Incremental distributed estimator for covariance.
|
|
29
|
+
Allows to distributely compute empirical covariance estimated by maximum
|
|
30
|
+
likelihood method if data are splitted into batches.
|
|
31
|
+
|
|
32
|
+
API is the same as for `sklearnex.covariance.IncrementalEmpiricalCovariance`
|
|
33
|
+
"""
|
|
34
|
+
|
|
35
|
+
_onedal_incremental_covariance = staticmethod(
|
|
36
|
+
onedalSPMD_IncrementalEmpiricalCovariance
|
|
37
|
+
)
|