scikit-learn-intelex 2023.2.1__py39-none-win_amd64.whl → 2024.0.1__py39-none-win_amd64.whl

scikit_learn_intelex-2023.2.1.data/data/Lib/site-packages/sklearnex/neighbors/knn_classification.py

@@ -1,228 +0,0 @@
-#!/usr/bin/env python
-#===============================================================================
-# Copyright 2021 Intel Corporation
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#===============================================================================
-
-from daal4py.sklearn._utils import sklearn_check_version
-import warnings
-
-from sklearn.neighbors._base import NeighborsBase as sklearn_NeighborsBase
-from sklearn.neighbors._ball_tree import BallTree
-from sklearn.neighbors._kd_tree import KDTree
-if not sklearn_check_version('1.2'):
-    from sklearn.neighbors._base import _check_weights
-from sklearn.neighbors._base import VALID_METRICS
-from sklearn.neighbors._classification import KNeighborsClassifier as \
-    sklearn_KNeighborsClassifier
-from sklearn.neighbors._unsupervised import NearestNeighbors as \
-    sklearn_NearestNeighbors
-from sklearn.utils.validation import _deprecate_positional_args, check_is_fitted
-
-from onedal.datatypes import _check_array, _num_features, _num_samples
-from onedal.neighbors import KNeighborsClassifier as onedal_KNeighborsClassifier
-
-from .common import KNeighborsDispatchingBase
-from .._device_offload import dispatch, wrap_output_data
-import numpy as np
-
-
-if sklearn_check_version("0.24"):
-    class KNeighborsClassifier_(sklearn_KNeighborsClassifier):
-        if sklearn_check_version('1.2'):
-            _parameter_constraints: dict = {
-                **sklearn_KNeighborsClassifier._parameter_constraints}
-
-        @_deprecate_positional_args
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None,
-                     **kwargs):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs, **kwargs)
-            self.weights = \
-                weights if sklearn_check_version("1.0") \
-                else _check_weights(weights)
-elif sklearn_check_version("0.22"):
-    from sklearn.neighbors._base import SupervisedIntegerMixin as \
-        BaseSupervisedIntegerMixin
-
-    class KNeighborsClassifier_(sklearn_KNeighborsClassifier,
-                                BaseSupervisedIntegerMixin):
-        @_deprecate_positional_args
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None,
-                     **kwargs):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs, **kwargs)
-            self.weights = _check_weights(weights)
-else:
-    from sklearn.neighbors.base import SupervisedIntegerMixin as \
-        BaseSupervisedIntegerMixin
-
-    class KNeighborsClassifier_(sklearn_KNeighborsClassifier,
-                                BaseSupervisedIntegerMixin):
-        @_deprecate_positional_args
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None,
-                     **kwargs):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs, **kwargs)
-            self.weights = _check_weights(weights)
-
-
-class KNeighborsClassifier(KNeighborsClassifier_, KNeighborsDispatchingBase):
-    if sklearn_check_version('1.2'):
-        _parameter_constraints: dict = {
-            **KNeighborsClassifier_._parameter_constraints}
-
-    if sklearn_check_version('1.0'):
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                weights=weights,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs)
-    else:
-        @_deprecate_positional_args
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None,
-                     **kwargs):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                weights=weights,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs, **kwargs)
-
-    def fit(self, X, y):
-        self._fit_validation(X, y)
-        dispatch(self, 'fit', {
-            'onedal': self.__class__._onedal_fit,
-            'sklearn': sklearn_KNeighborsClassifier.fit,
-        }, X, y)
-        return self
-
-    @wrap_output_data
-    def predict(self, X):
-        check_is_fitted(self)
-        if sklearn_check_version("1.0"):
-            self._check_feature_names(X, reset=False)
-        return dispatch(self, 'predict', {
-            'onedal': self.__class__._onedal_predict,
-            'sklearn': sklearn_KNeighborsClassifier.predict,
-        }, X)
-
-    @wrap_output_data
-    def predict_proba(self, X):
-        check_is_fitted(self)
-        if sklearn_check_version("1.0"):
-            self._check_feature_names(X, reset=False)
-        return dispatch(self, 'predict_proba', {
-            'onedal': self.__class__._onedal_predict_proba,
-            'sklearn': sklearn_KNeighborsClassifier.predict_proba,
-        }, X)
-
-    @wrap_output_data
-    def kneighbors(self, X=None, n_neighbors=None, return_distance=True):
-        check_is_fitted(self)
-        if sklearn_check_version("1.0"):
-            self._check_feature_names(X, reset=False)
-        return dispatch(self, 'kneighbors', {
-            'onedal': self.__class__._onedal_kneighbors,
-            'sklearn': sklearn_KNeighborsClassifier.kneighbors,
-        }, X, n_neighbors, return_distance)
-
-    @wrap_output_data
-    def radius_neighbors(self, X=None, radius=None, return_distance=True,
-                         sort_results=False):
-        _onedal_estimator = getattr(self, '_onedal_estimator', None)
-
-        if _onedal_estimator is not None or getattr(self, '_tree', 0) is None and \
-                self._fit_method == 'kd_tree':
-            if sklearn_check_version("0.24"):
-                sklearn_NearestNeighbors.fit(self, self._fit_X, getattr(self, '_y', None))
-            else:
-                sklearn_NearestNeighbors.fit(self, self._fit_X)
-        if sklearn_check_version("0.22"):
-            result = sklearn_NearestNeighbors.radius_neighbors(
-                self, X, radius, return_distance, sort_results)
-        else:
-            result = sklearn_NearestNeighbors.radius_neighbors(
-                self, X, radius, return_distance)
-
-        return result
-
-    def _onedal_fit(self, X, y, queue=None):
-        onedal_params = {
-            'n_neighbors': self.n_neighbors,
-            'weights': self.weights,
-            'algorithm': self.algorithm,
-            'metric': self.effective_metric_,
-            'p': self.effective_metric_params_['p'],
-        }
-
-        try:
-            requires_y = self._get_tags()["requires_y"]
-        except KeyError:
-            requires_y = False
-
-        self._onedal_estimator = onedal_KNeighborsClassifier(**onedal_params)
-        self._onedal_estimator.requires_y = requires_y
-        self._onedal_estimator.effective_metric_ = self.effective_metric_
-        self._onedal_estimator.effective_metric_params_ = self.effective_metric_params_
-        self._onedal_estimator.fit(X, y, queue=queue)
-
-        self._save_attributes()
-
-    def _onedal_predict(self, X, queue=None):
-        return self._onedal_estimator.predict(X, queue=queue)
-
-    def _onedal_predict_proba(self, X, queue=None):
-        return self._onedal_estimator.predict_proba(X, queue=queue)
-
-    def _onedal_kneighbors(self, X=None, n_neighbors=None,
-                           return_distance=True, queue=None):
-        return self._onedal_estimator.kneighbors(
-            X, n_neighbors, return_distance, queue=queue)
-
-    def _save_attributes(self):
-        self.classes_ = self._onedal_estimator.classes_
-        self.n_features_in_ = self._onedal_estimator.n_features_in_
-        self.n_samples_fit_ = self._onedal_estimator.n_samples_fit_
-        self._fit_X = self._onedal_estimator._fit_X
-        self._y = self._onedal_estimator._y
-        self._fit_method = self._onedal_estimator._fit_method
-        self.outputs_2d_ = self._onedal_estimator.outputs_2d_
-        self._tree = self._onedal_estimator._tree

scikit_learn_intelex-2023.2.1.data/data/Lib/site-packages/sklearnex/neighbors/knn_regression.py

@@ -1,213 +0,0 @@
-#!/usr/bin/env python
-#===============================================================================
-# Copyright 2021 Intel Corporation
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#===============================================================================
-
-from daal4py.sklearn._utils import sklearn_check_version
-import warnings
-
-from sklearn.neighbors._base import NeighborsBase as sklearn_NeighborsBase
-from sklearn.neighbors._ball_tree import BallTree
-from sklearn.neighbors._kd_tree import KDTree
-if not sklearn_check_version('1.2'):
-    from sklearn.neighbors._base import _check_weights
-from sklearn.neighbors._base import VALID_METRICS
-from sklearn.neighbors._regression import KNeighborsRegressor as \
-    sklearn_KNeighborsRegressor
-from sklearn.neighbors._unsupervised import NearestNeighbors as \
-    sklearn_NearestNeighbors
-from sklearn.utils.validation import _deprecate_positional_args, check_is_fitted
-
-from onedal.datatypes import _check_array, _num_features, _num_samples
-from onedal.neighbors import KNeighborsRegressor as onedal_KNeighborsRegressor
-
-from .common import KNeighborsDispatchingBase
-from .._device_offload import dispatch, wrap_output_data
-import numpy as np
-
-
-if sklearn_check_version("0.24"):
-    class KNeighborsRegressor_(sklearn_KNeighborsRegressor):
-        if sklearn_check_version('1.2'):
-            _parameter_constraints: dict = {
-                **sklearn_KNeighborsRegressor._parameter_constraints}
-
-        @_deprecate_positional_args
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None,
-                     **kwargs):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs, **kwargs)
-            self.weights = \
-                weights if sklearn_check_version("1.0") \
-                else _check_weights(weights)
-elif sklearn_check_version("0.22"):
-    from sklearn.neighbors._base import SupervisedFloatMixin as \
-        BaseSupervisedFloatMixin
-
-    class KNeighborsRegressor_(sklearn_KNeighborsRegressor,
-                               BaseSupervisedFloatMixin):
-        @_deprecate_positional_args
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None,
-                     **kwargs):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs, **kwargs)
-            self.weights = _check_weights(weights)
-else:
-    from sklearn.neighbors.base import SupervisedFloatMixin as \
-        BaseSupervisedFloatMixin
-
-    class KNeighborsRegressor_(sklearn_KNeighborsRegressor,
-                               BaseSupervisedFloatMixin):
-        @_deprecate_positional_args
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None,
-                     **kwargs):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs, **kwargs)
-            self.weights = _check_weights(weights)
-
-
-class KNeighborsRegressor(KNeighborsRegressor_, KNeighborsDispatchingBase):
-    if sklearn_check_version('1.2'):
-        _parameter_constraints: dict = {
-            **KNeighborsRegressor_._parameter_constraints}
-
-    if sklearn_check_version('1.0'):
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                weights=weights,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs)
-    else:
-        @_deprecate_positional_args
-        def __init__(self, n_neighbors=5, *,
-                     weights='uniform', algorithm='auto', leaf_size=30,
-                     p=2, metric='minkowski', metric_params=None, n_jobs=None,
-                     **kwargs):
-            super().__init__(
-                n_neighbors=n_neighbors,
-                weights=weights,
-                algorithm=algorithm,
-                leaf_size=leaf_size, metric=metric, p=p,
-                metric_params=metric_params,
-                n_jobs=n_jobs, **kwargs)
-
-    def fit(self, X, y):
-        self._fit_validation(X, y)
-        dispatch(self, 'fit', {
-            'onedal': self.__class__._onedal_fit,
-            'sklearn': sklearn_KNeighborsRegressor.fit,
-        }, X, y)
-        return self
-
-    @wrap_output_data
-    def predict(self, X):
-        check_is_fitted(self)
-        if sklearn_check_version("1.0"):
-            self._check_feature_names(X, reset=False)
-        return dispatch(self, 'predict', {
-            'onedal': self.__class__._onedal_predict,
-            'sklearn': sklearn_KNeighborsRegressor.predict,
-        }, X)
-
-    @wrap_output_data
-    def kneighbors(self, X=None, n_neighbors=None, return_distance=True):
-        check_is_fitted(self)
-        if sklearn_check_version("1.0"):
-            self._check_feature_names(X, reset=False)
-        return dispatch(self, 'kneighbors', {
-            'onedal': self.__class__._onedal_kneighbors,
-            'sklearn': sklearn_KNeighborsRegressor.kneighbors,
-        }, X, n_neighbors, return_distance)
-
-    @wrap_output_data
-    def radius_neighbors(self, X=None, radius=None, return_distance=True,
-                         sort_results=False):
-        _onedal_estimator = getattr(self, '_onedal_estimator', None)
-
-        if _onedal_estimator is not None or getattr(self, '_tree', 0) is None and \
-                self._fit_method == 'kd_tree':
-            if sklearn_check_version("0.24"):
-                sklearn_NearestNeighbors.fit(self, self._fit_X, getattr(self, '_y', None))
-            else:
-                sklearn_NearestNeighbors.fit(self, self._fit_X)
-        if sklearn_check_version("0.22"):
-            result = sklearn_NearestNeighbors.radius_neighbors(
-                self, X, radius, return_distance, sort_results)
-        else:
-            result = sklearn_NearestNeighbors.radius_neighbors(
-                self, X, radius, return_distance)
-
-        return result
-
-    def _onedal_fit(self, X, y, queue=None):
-        onedal_params = {
-            'n_neighbors': self.n_neighbors,
-            'weights': self.weights,
-            'algorithm': self.algorithm,
-            'metric': self.effective_metric_,
-            'p': self.effective_metric_params_['p'],
-        }
-
-        try:
-            requires_y = self._get_tags()["requires_y"]
-        except KeyError:
-            requires_y = False
-
-        self._onedal_estimator = onedal_KNeighborsRegressor(**onedal_params)
-        self._onedal_estimator.requires_y = requires_y
-        self._onedal_estimator.effective_metric_ = self.effective_metric_
-        self._onedal_estimator.effective_metric_params_ = self.effective_metric_params_
-        self._onedal_estimator.fit(X, y, queue=queue)
-
-        self._save_attributes()
-
-    def _onedal_predict(self, X, queue=None):
-        return self._onedal_estimator.predict(X, queue=queue)
-
-    def _onedal_kneighbors(self, X=None, n_neighbors=None,
-                           return_distance=True, queue=None):
-        return self._onedal_estimator.kneighbors(
-            X, n_neighbors, return_distance, queue=queue)
-
-    def _save_attributes(self):
-        self.n_features_in_ = self._onedal_estimator.n_features_in_
-        self.n_samples_fit_ = self._onedal_estimator.n_samples_fit_
-        self._fit_X = self._onedal_estimator._fit_X
-        self._y = self._onedal_estimator._y
-        self._fit_method = self._onedal_estimator._fit_method
-        self._tree = self._onedal_estimator._tree

scikit_learn_intelex-2023.2.1.data/data/Lib/site-packages/sklearnex/neighbors/tests/test_neighbors.py

@@ -1,57 +0,0 @@
-#!/usr/bin/env python
-#===============================================================================
-# Copyright 2021 Intel Corporation
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#===============================================================================
-
-import numpy as np
-from numpy.testing import assert_allclose
-
-
-def test_sklearnex_import_knn_classifier():
-    from sklearnex.neighbors import KNeighborsClassifier
-    X = [[0], [1], [2], [3]]
-    y = [0, 0, 1, 1]
-    neigh = KNeighborsClassifier(n_neighbors=3).fit(X, y)
-    assert 'sklearnex' in neigh.__module__
-    assert_allclose(neigh.predict([[1.1]]), [0])
-
-
-def test_sklearnex_import_knn_regression():
-    from sklearnex.neighbors import KNeighborsRegressor
-    X = [[0], [1], [2], [3]]
-    y = [0, 0, 1, 1]
-    neigh = KNeighborsRegressor(n_neighbors=2).fit(X, y)
-    assert 'sklearnex' in neigh.__module__
-    assert_allclose(neigh.predict([[1.5]]), [0.5])
-
-
-def test_sklearnex_import_nn():
-    from sklearnex.neighbors import NearestNeighbors
-    X = [[0, 0, 2], [1, 0, 0], [0, 0, 1]]
-    neigh = NearestNeighbors(n_neighbors=2).fit(X)
-    assert 'sklearnex' in neigh.__module__
-    result = neigh.kneighbors([[0, 0, 1.3]], 2, return_distance=False)
-    assert_allclose(result, [[2, 0]])
-
-
-def test_sklearnex_import_lof():
-    from sklearnex.neighbors import LocalOutlierFactor
-    X = [[7, 7, 7], [1, 0, 0], [0, 0, 1], [0, 0, 1]]
-    lof = LocalOutlierFactor(n_neighbors=2)
-    result = lof.fit_predict(X)
-    assert hasattr(lof, '_knn')
-    assert 'sklearnex' in lof.__module__
-    assert 'sklearnex' in lof._knn.__module__
-    assert_allclose(result, [-1, 1, 1, 1])

scikit_learn_intelex-2023.2.1.data/data/Lib/site-packages/sklearnex/preview/__init__.py

@@ -1,18 +0,0 @@
-#!/usr/bin/env python
-#===============================================================================
-# Copyright 2023 Intel Corporation
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#===============================================================================
-
-__all__ = ['cluster', 'decomposition', 'linear_model', 'ensemble']

scikit_learn_intelex-2023.2.1.data/data/Lib/site-packages/sklearnex/preview/decomposition/tests/test_preview_pca.py

@@ -1,28 +0,0 @@
-#!/usr/bin/env python
-#===============================================================================
-# Copyright 2023 Intel Corporation
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#===============================================================================
-
-import numpy as np
-from numpy.testing import assert_allclose
-
-
-def test_sklearnex_import():
-    from sklearnex.preview.decomposition import PCA
-    X = np.array([[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]])
-    pca = PCA(n_components=2, svd_solver='full').fit(X)
-    assert 'sklearnex' in pca.__module__
-    assert hasattr(pca, '_onedal_estimator')
-    assert_allclose(pca.singular_values_, [6.30061232, 0.54980396])