scikit-learn-intelex 2024.1.0__py311-none-win_amd64.whl → 2025.1.0__py311-none-win_amd64.whl

scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_linear.py

@@ -0,0 +1,207 @@
+# ===============================================================================
+# Copyright 2024 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
+import pytest
+from numpy.testing import assert_allclose
+
+from onedal.tests.utils._dataframes_support import (
+    _as_numpy,
+    _convert_to_dataframe,
+    get_dataframes_and_queues,
+)
+from sklearnex.linear_model import IncrementalLinearRegression
+from sklearnex.tests.utils import _IS_INTEL
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+@pytest.mark.parametrize("fit_intercept", [True, False])
+@pytest.mark.parametrize("macro_block", [None, 1024])
+@pytest.mark.parametrize("dtype", [np.float32, np.float64])
+def test_sklearnex_fit_on_gold_data(dataframe, queue, fit_intercept, macro_block, dtype):
+    X = np.array([[1], [2]])
+    X = X.astype(dtype=dtype)
+    X_df = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+    y = np.array([[1], [2]])
+    y = y.astype(dtype=dtype)
+    y_df = _convert_to_dataframe(y, sycl_queue=queue, target_df=dataframe)
+
+    inclin = IncrementalLinearRegression(fit_intercept=fit_intercept)
+    if macro_block is not None:
+        hparams = inclin.get_hyperparameters("fit")
+        hparams.cpu_macro_block = macro_block
+        hparams.gpu_macro_block = macro_block
+    inclin.fit(X_df, y_df)
+
+    y_pred = inclin.predict(X_df)
+    np_y_pred = _as_numpy(y_pred)
+
+    tol = 2e-6 if dtype == np.float32 else 1e-7
+    assert_allclose(inclin.coef_, [1], atol=tol)
+    if fit_intercept:
+        assert_allclose(inclin.intercept_, [0], atol=tol)
+    assert_allclose(np_y_pred, y, atol=tol)
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+@pytest.mark.parametrize("fit_intercept", [True, False])
+@pytest.mark.parametrize("macro_block", [None, 1024])
+@pytest.mark.parametrize("dtype", [np.float32, np.float64])
+def test_sklearnex_partial_fit_on_gold_data(
+    dataframe, queue, fit_intercept, macro_block, dtype
+):
+    X = np.array([[1], [2], [3], [4]])
+    X = X.astype(dtype=dtype)
+    y = X + 3
+    y = y.astype(dtype=dtype)
+    X_split = np.array_split(X, 2)
+    y_split = np.array_split(y, 2)
+
+    inclin = IncrementalLinearRegression()
+    if macro_block is not None:
+        hparams = inclin.get_hyperparameters("fit")
+        hparams.cpu_macro_block = macro_block
+        hparams.gpu_macro_block = macro_block
+    for i in range(2):
+        X_split_df = _convert_to_dataframe(
+            X_split[i], sycl_queue=queue, target_df=dataframe
+        )
+        y_split_df = _convert_to_dataframe(
+            y_split[i], sycl_queue=queue, target_df=dataframe
+        )
+        inclin.partial_fit(X_split_df, y_split_df)
+
+    X_df = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+    y_pred = inclin.predict(X_df)
+    np_y_pred = _as_numpy(y_pred)
+
+    assert inclin.n_features_in_ == 1
+    tol = 1e-5 if dtype == np.float32 else 1e-7
+    assert_allclose(inclin.coef_, [[1]], atol=tol)
+    if fit_intercept:
+        assert_allclose(inclin.intercept_, 3, atol=tol)
+
+    assert_allclose(np_y_pred, y, atol=tol)
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+@pytest.mark.parametrize("fit_intercept", [True, False])
+@pytest.mark.parametrize("macro_block", [None, 1024])
+@pytest.mark.parametrize("dtype", [np.float32, np.float64])
+def test_sklearnex_partial_fit_multitarget_on_gold_data(
+    dataframe, queue, fit_intercept, macro_block, dtype
+):
+    X = np.array([[1, 1], [1, 2], [2, 2], [2, 3]])
+    X = X.astype(dtype=dtype)
+    y = np.dot(X, [1, 2]) + 3
+    y = y.astype(dtype=dtype)
+    X_split = np.array_split(X, 2)
+    y_split = np.array_split(y, 2)
+
+    inclin = IncrementalLinearRegression()
+    if macro_block is not None:
+        hparams = inclin.get_hyperparameters("fit")
+        hparams.cpu_macro_block = macro_block
+        hparams.gpu_macro_block = macro_block
+    for i in range(2):
+        X_split_df = _convert_to_dataframe(
+            X_split[i], sycl_queue=queue, target_df=dataframe
+        )
+        y_split_df = _convert_to_dataframe(
+            y_split[i], sycl_queue=queue, target_df=dataframe
+        )
+        inclin.partial_fit(X_split_df, y_split_df)
+
+    X_df = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+    y_pred = inclin.predict(X_df)
+    np_y_pred = _as_numpy(y_pred)
+
+    assert inclin.n_features_in_ == 2
+    tol = 1e-7
+    if dtype == np.float32:
+        tol = 7e-6 if _IS_INTEL else 2e-5
+
+    assert_allclose(inclin.coef_, [1.0, 2.0], atol=tol)
+    if fit_intercept:
+        assert_allclose(inclin.intercept_, 3.0, atol=tol)
+
+    assert_allclose(np_y_pred, y, atol=tol)
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+@pytest.mark.parametrize("fit_intercept", [True, False])
+@pytest.mark.parametrize("num_samples", [100, 1000])
+@pytest.mark.parametrize("num_features", [5, 10])
+@pytest.mark.parametrize("num_targets", [1, 2])
+@pytest.mark.parametrize("num_blocks", [1, 10])
+@pytest.mark.parametrize("macro_block", [None, 1024])
+@pytest.mark.parametrize("dtype", [np.float32, np.float64])
+def test_sklearnex_partial_fit_on_random_data(
+    dataframe,
+    queue,
+    fit_intercept,
+    num_samples,
+    num_features,
+    num_targets,
+    num_blocks,
+    macro_block,
+    dtype,
+):
+    seed = 42
+    gen = np.random.default_rng(seed)
+    intercept = gen.random(size=num_targets, dtype=dtype)
+    coef = gen.random(size=(num_targets, num_features), dtype=dtype).T
+
+    X = gen.random(size=(num_samples, num_features), dtype=dtype)
+    if fit_intercept:
+        y = X @ coef + intercept[np.newaxis, :]
+    else:
+        y = X @ coef
+
+    X_split = np.array_split(X, num_blocks)
+    y_split = np.array_split(y, num_blocks)
+
+    inclin = IncrementalLinearRegression(fit_intercept=fit_intercept)
+    if macro_block is not None:
+        hparams = inclin.get_hyperparameters("fit")
+        hparams.cpu_macro_block = macro_block
+        hparams.gpu_macro_block = macro_block
+    for i in range(num_blocks):
+        X_split_df = _convert_to_dataframe(
+            X_split[i], sycl_queue=queue, target_df=dataframe
+        )
+        y_split_df = _convert_to_dataframe(
+            y_split[i], sycl_queue=queue, target_df=dataframe
+        )
+        inclin.partial_fit(X_split_df, y_split_df)
+
+    tol = 1e-4 if inclin.coef_.dtype == np.float32 else 1e-7
+    assert_allclose(coef.T.squeeze(), inclin.coef_, atol=tol)
+
+    if fit_intercept:
+        assert_allclose(intercept, inclin.intercept_, atol=tol)
+
+    X_test = gen.random(size=(num_samples, num_features), dtype=dtype)
+    if fit_intercept:
+        expected_y_pred = (X_test @ coef + intercept[np.newaxis, :]).squeeze()
+    else:
+        expected_y_pred = (X_test @ coef).squeeze()
+
+    X_test_df = _convert_to_dataframe(X_test, sycl_queue=queue, target_df=dataframe)
+
+    y_pred = inclin.predict(X_test_df)
+
+    assert_allclose(expected_y_pred, _as_numpy(y_pred), atol=tol)

scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_incremental_ridge.py

@@ -0,0 +1,153 @@
+# ===============================================================================
+# Copyright 2024 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 daal_check_version
+
+if daal_check_version((2024, "P", 600)):
+    import numpy as np
+    import pytest
+    from numpy.testing import assert_allclose
+    from sklearn.exceptions import NotFittedError
+
+    from onedal.tests.utils._dataframes_support import (
+        _as_numpy,
+        _convert_to_dataframe,
+        get_dataframes_and_queues,
+    )
+    from sklearnex.linear_model import IncrementalRidge
+
+    def _compute_ridge_coefficients(X, y, alpha, fit_intercept):
+        coefficients_manual, intercept_manual = None, None
+        if fit_intercept:
+            X_mean = np.mean(X, axis=0)
+            y_mean = np.mean(y)
+            X_centered = X - X_mean
+            y_centered = y - y_mean
+
+            X_with_intercept = np.hstack([np.ones((X.shape[0], 1)), X_centered])
+            lambda_identity = alpha * np.eye(X_with_intercept.shape[1])
+            inverse_term = np.linalg.inv(
+                np.dot(X_with_intercept.T, X_with_intercept) + lambda_identity
+            )
+            xt_y = np.dot(X_with_intercept.T, y_centered)
+            coefficients_manual = np.dot(inverse_term, xt_y)
+
+            intercept_manual = y_mean - np.dot(X_mean, coefficients_manual[1:])
+            coefficients_manual = coefficients_manual[1:]
+        else:
+            lambda_identity = alpha * np.eye(X.shape[1])
+            inverse_term = np.linalg.inv(np.dot(X.T, X) + lambda_identity)
+            xt_y = np.dot(X.T, y)
+            coefficients_manual = np.dot(inverse_term, xt_y)
+
+        return coefficients_manual, intercept_manual
+
+    @pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+    @pytest.mark.parametrize("batch_size", [10, 100, 1000])
+    @pytest.mark.parametrize("alpha", [0.1, 0.5, 1.0])
+    @pytest.mark.parametrize("fit_intercept", [True, False])
+    def test_inc_ridge_fit_coefficients(
+        dataframe, queue, alpha, batch_size, fit_intercept
+    ):
+        sample_size, feature_size = 1000, 50
+        X = np.random.rand(sample_size, feature_size)
+        y = np.random.rand(sample_size)
+        X_c = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+        y_c = _convert_to_dataframe(y, sycl_queue=queue, target_df=dataframe)
+
+        inc_ridge = IncrementalRidge(
+            fit_intercept=fit_intercept, alpha=alpha, batch_size=batch_size
+        )
+        inc_ridge.fit(X_c, y_c)
+
+        coefficients_manual, intercept_manual = _compute_ridge_coefficients(
+            X, y, alpha, fit_intercept
+        )
+        if fit_intercept:
+            assert_allclose(inc_ridge.intercept_, intercept_manual, rtol=1e-6, atol=1e-6)
+
+        assert_allclose(inc_ridge.coef_, coefficients_manual, rtol=1e-6, atol=1e-6)
+
+    @pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+    @pytest.mark.parametrize("batch_size", [2, 5])
+    @pytest.mark.parametrize("alpha", [0.1, 0.5, 1.0])
+    def test_inc_ridge_partial_fit_coefficients(dataframe, queue, alpha, batch_size):
+        sample_size, feature_size = 1000, 50
+        X = np.random.rand(sample_size, feature_size)
+        y = np.random.rand(sample_size)
+        X_split = np.array_split(X, batch_size)
+        y_split = np.array_split(y, batch_size)
+
+        inc_ridge = IncrementalRidge(fit_intercept=False, alpha=alpha)
+
+        for batch_index in range(len(X_split)):
+            X_c = _convert_to_dataframe(
+                X_split[batch_index], sycl_queue=queue, target_df=dataframe
+            )
+            y_c = _convert_to_dataframe(
+                y_split[batch_index], sycl_queue=queue, target_df=dataframe
+            )
+            inc_ridge.partial_fit(X_c, y_c)
+
+        lambda_identity = alpha * np.eye(X.shape[1])
+        inverse_term = np.linalg.inv(np.dot(X.T, X) + lambda_identity)
+        xt_y = np.dot(X.T, y)
+        coefficients_manual = np.dot(inverse_term, xt_y)
+
+        assert_allclose(inc_ridge.coef_, coefficients_manual, rtol=1e-6, atol=1e-6)
+
+    def test_inc_ridge_score_before_fit():
+        X = np.array([[1, 1], [1, 2], [2, 2], [2, 3]])
+        y = np.dot(X, np.array([1, 2])) + 3
+        inc_ridge = IncrementalRidge(alpha=0.5)
+        with pytest.raises(NotFittedError):
+            inc_ridge.score(X, y)
+
+    def test_inc_ridge_predict_before_fit():
+        X = np.array([[1, 1], [1, 2], [2, 2], [2, 3]])
+        inc_ridge = IncrementalRidge(alpha=0.5)
+        with pytest.raises(NotFittedError):
+            inc_ridge.predict(X)
+
+    def test_inc_ridge_score_after_fit():
+        X = np.array([[1, 1], [1, 2], [2, 2], [2, 3]])
+        y = np.dot(X, np.array([1, 2])) + 3
+        inc_ridge = IncrementalRidge(alpha=0.5)
+        inc_ridge.fit(X, y)
+        assert inc_ridge.score(X, y) >= 0.97
+
+    @pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+    @pytest.mark.parametrize("fit_intercept", [True, False])
+    def test_inc_ridge_predict_after_fit(dataframe, queue, fit_intercept):
+        sample_size, feature_size = 1000, 50
+        X = np.random.rand(sample_size, feature_size)
+        y = np.random.rand(sample_size)
+        X_c = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+        y_c = _convert_to_dataframe(y, sycl_queue=queue, target_df=dataframe)
+
+        inc_ridge = IncrementalRidge(fit_intercept=fit_intercept, alpha=0.5)
+        inc_ridge.fit(X_c, y_c)
+
+        y_pred = inc_ridge.predict(X_c)
+
+        coefficients_manual, intercept_manual = _compute_ridge_coefficients(
+            X, y, 0.5, fit_intercept
+        )
+        y_pred_manual = np.dot(X, coefficients_manual)
+        if fit_intercept:
+            y_pred_manual += intercept_manual
+
+        assert_allclose(_as_numpy(y_pred), y_pred_manual, rtol=1e-6, atol=1e-6)

scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_linear.py

@@ -0,0 +1,167 @@
+# ===============================================================================
+# 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
+import pytest
+from numpy.testing import assert_allclose
+from scipy.linalg import lstsq
+from sklearn.datasets import make_regression
+
+from daal4py.sklearn._utils import daal_check_version
+from daal4py.sklearn.linear_model.tests.test_ridge import (
+    _test_multivariate_ridge_alpha_shape,
+    _test_multivariate_ridge_coefficients,
+)
+from onedal.tests.utils._dataframes_support import (
+    _as_numpy,
+    _convert_to_dataframe,
+    get_dataframes_and_queues,
+)
+from sklearnex.tests.utils import _IS_INTEL
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+@pytest.mark.parametrize("dtype", [np.float32, np.float64])
+@pytest.mark.parametrize("macro_block", [None, 1024])
+@pytest.mark.parametrize("overdetermined", [False, True])
+@pytest.mark.parametrize("multi_output", [False, True])
+def test_sklearnex_import_linear(
+    dataframe, queue, dtype, macro_block, overdetermined, multi_output
+):
+    if (overdetermined or multi_output) and not daal_check_version((2025, "P", 1)):
+        pytest.skip()
+    if overdetermined and queue and queue.sycl_device.is_gpu:
+        pytest.skip()
+
+    from sklearnex.linear_model import LinearRegression
+
+    rng = np.random.default_rng(seed=123)
+    X = rng.standard_normal(size=(10, 20) if overdetermined else (20, 5))
+    y = rng.standard_normal(size=(X.shape[0], 3) if multi_output else X.shape[0])
+
+    Xi = np.c_[X, np.ones((X.shape[0], 1))]
+    expected_coefs = lstsq(Xi, y)[0]
+    expected_intercept = expected_coefs[-1]
+    expected_coefs = expected_coefs[: X.shape[1]]
+    if multi_output:
+        expected_coefs = expected_coefs.T
+
+    linreg = LinearRegression()
+    if daal_check_version((2024, "P", 0)) and macro_block is not None:
+        hparams = linreg.get_hyperparameters("fit")
+        hparams.cpu_macro_block = macro_block
+        hparams.gpu_macro_block = macro_block
+
+    X = X.astype(dtype=dtype)
+    y = y.astype(dtype=dtype)
+    y_list = y.tolist()
+    X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+    y = _convert_to_dataframe(y, sycl_queue=queue, target_df=dataframe)
+    linreg.fit(X, y)
+
+    assert hasattr(linreg, "_onedal_estimator")
+    assert "sklearnex" in linreg.__module__
+
+    rtol = 1e-3 if dtype == np.float32 else 1e-5
+    assert_allclose(_as_numpy(linreg.coef_), expected_coefs, rtol=rtol)
+    assert_allclose(_as_numpy(linreg.intercept_), expected_intercept, rtol=rtol)
+
+    # check that it also works with lists
+    if isinstance(X, np.ndarray):
+        linreg_list = LinearRegression().fit(X, y_list)
+        assert_allclose(linreg_list.coef_, linreg.coef_)
+        assert_allclose(linreg_list.intercept_, linreg.intercept_)
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+def test_sklearnex_import_ridge(dataframe, queue):
+    from sklearnex.linear_model import Ridge
+
+    X = np.array([[1, 1], [1, 2], [2, 2], [2, 3]])
+    y = np.dot(X, np.array([1, 2])) + 3
+    X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+    y = _convert_to_dataframe(y, sycl_queue=queue, target_df=dataframe)
+    ridgereg = Ridge().fit(X, y)
+    assert "daal4py" in ridgereg.__module__
+    assert_allclose(ridgereg.intercept_, 4.5)
+    assert_allclose(ridgereg.coef_, [0.8, 1.4])
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+def test_sklearnex_import_lasso(dataframe, queue):
+    from sklearnex.linear_model import Lasso
+
+    X = [[0, 0], [1, 1], [2, 2]]
+    y = [0, 1, 2]
+    X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+    y = _convert_to_dataframe(y, sycl_queue=queue, target_df=dataframe)
+    lasso = Lasso(alpha=0.1).fit(X, y)
+    assert "daal4py" in lasso.__module__
+    assert_allclose(lasso.intercept_, 0.15)
+    assert_allclose(lasso.coef_, [0.85, 0.0])
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+def test_sklearnex_import_elastic(dataframe, queue):
+    from sklearnex.linear_model import ElasticNet
+
+    X, y = make_regression(n_features=2, random_state=0)
+    X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+    y = _convert_to_dataframe(y, sycl_queue=queue, target_df=dataframe)
+    elasticnet = ElasticNet(random_state=0).fit(X, y)
+    assert "daal4py" in elasticnet.__module__
+    assert_allclose(elasticnet.intercept_, 1.451, atol=1e-3)
+    assert_allclose(elasticnet.coef_, [18.838, 64.559], atol=1e-3)
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+@pytest.mark.parametrize("dtype", [np.float32, np.float64])
+def test_sklearnex_reconstruct_model(dataframe, queue, dtype):
+    from sklearnex.linear_model import LinearRegression
+
+    seed = 42
+    num_samples = 3500
+    num_features, num_targets = 14, 9
+
+    gen = np.random.default_rng(seed)
+    intercept = gen.random(size=num_targets, dtype=dtype)
+    coef = gen.random(size=(num_targets, num_features), dtype=dtype).T
+
+    X = gen.random(size=(num_samples, num_features), dtype=dtype)
+    gtr = X @ coef + intercept[np.newaxis, :]
+
+    X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
+
+    linreg = LinearRegression(fit_intercept=True)
+    linreg.coef_ = coef.T
+    linreg.intercept_ = intercept
+
+    y_pred = linreg.predict(X)
+
+    tol = 1e-5 if _as_numpy(y_pred).dtype == np.float32 else 1e-7
+    assert_allclose(gtr, _as_numpy(y_pred), rtol=tol)
+
+
+def test_sklearnex_multivariate_ridge_coefs():
+    from sklearnex.linear_model import Ridge
+
+    _test_multivariate_ridge_coefficients(Ridge, random_state=0)
+
+
+def test_sklearnex_multivariate_ridge_alpha_shape():
+    from sklearnex.linear_model import Ridge
+
+    _test_multivariate_ridge_alpha_shape(Ridge, random_state=0)

scikit_learn_intelex-2025.1.0.data/data/Lib/site-packages/sklearnex/linear_model/tests/test_logreg.py

@@ -0,0 +1,134 @@
+# ===============================================================================
+# 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
+import pytest
+from numpy.testing import assert_allclose, assert_array_equal
+from scipy.sparse import csr_matrix
+from sklearn.datasets import load_breast_cancer, load_iris, make_classification
+from sklearn.metrics import accuracy_score
+from sklearn.model_selection import train_test_split
+
+from daal4py.sklearn._utils import daal_check_version
+from onedal.tests.utils._dataframes_support import (
+    _as_numpy,
+    _convert_to_dataframe,
+    get_dataframes_and_queues,
+    get_queues,
+)
+from sklearnex import config_context
+
+
+def prepare_input(X, y, dataframe, queue):
+    X_train, X_test, y_train, y_test = train_test_split(
+        X, y, train_size=0.8, random_state=42
+    )
+    X_train = _convert_to_dataframe(X_train, sycl_queue=queue, target_df=dataframe)
+    y_train = _convert_to_dataframe(y_train, sycl_queue=queue, target_df=dataframe)
+    X_test = _convert_to_dataframe(X_test, sycl_queue=queue, target_df=dataframe)
+    return X_train, X_test, y_train, y_test
+
+
+@pytest.mark.parametrize(
+    "dataframe,queue", get_dataframes_and_queues(device_filter_="cpu")
+)
+def test_sklearnex_multiclass_classification(dataframe, queue):
+    from sklearnex.linear_model import LogisticRegression
+
+    X, y = load_iris(return_X_y=True)
+    X_train, X_test, y_train, y_test = prepare_input(X, y, dataframe, queue)
+
+    logreg = LogisticRegression(fit_intercept=True, solver="lbfgs", max_iter=200).fit(
+        X_train, y_train
+    )
+
+    if daal_check_version((2024, "P", 1)):
+        assert "sklearnex" in logreg.__module__
+    else:
+        assert "daal4py" in logreg.__module__
+
+    y_pred = _as_numpy(logreg.predict(X_test))
+    assert accuracy_score(y_test, y_pred) > 0.99
+
+
+@pytest.mark.parametrize(
+    "dataframe,queue",
+    get_dataframes_and_queues(),
+)
+def test_sklearnex_binary_classification(dataframe, queue):
+    from sklearnex.linear_model import LogisticRegression
+
+    X, y = load_breast_cancer(return_X_y=True)
+    X_train, X_test, y_train, y_test = prepare_input(X, y, dataframe, queue)
+
+    logreg = LogisticRegression(fit_intercept=True, solver="newton-cg", max_iter=100).fit(
+        X_train, y_train
+    )
+
+    if daal_check_version((2024, "P", 1)):
+        assert "sklearnex" in logreg.__module__
+    else:
+        assert "daal4py" in logreg.__module__
+    if (
+        dataframe != "numpy"
+        and queue is not None
+        and queue.sycl_device.is_gpu
+        and daal_check_version((2024, "P", 1))
+    ):
+        # fit was done on gpu
+        assert hasattr(logreg, "_onedal_estimator")
+
+    y_pred = _as_numpy(logreg.predict(X_test))
+    assert accuracy_score(y_test, y_pred) > 0.95
+
+
+if daal_check_version((2024, "P", 700)):
+
+    @pytest.mark.parametrize("queue", get_queues("gpu"))
+    @pytest.mark.parametrize("dtype", [np.float32, np.float64])
+    @pytest.mark.parametrize(
+        "dims", [(3007, 17, 0.05), (50000, 100, 0.01), (512, 10, 0.5)]
+    )
+    def test_csr(queue, dtype, dims):
+        from sklearnex.linear_model import LogisticRegression
+
+        n, p, density = dims
+
+        # Create sparse dataset for classification
+        X, y = make_classification(n, p, random_state=42)
+        X = X.astype(dtype)
+        y = y.astype(dtype)
+        np.random.seed(2007 + n + p)
+        mask = np.random.binomial(1, density, (n, p))
+        X = X * mask
+        X_sp = csr_matrix(X)
+
+        model = LogisticRegression(fit_intercept=True, solver="newton-cg")
+        model_sp = LogisticRegression(fit_intercept=True, solver="newton-cg")
+
+        with config_context(target_offload="gpu:0"):
+            model.fit(X, y)
+            pred = model.predict(X)
+            prob = model.predict_proba(X)
+            model_sp.fit(X_sp, y)
+            pred_sp = model_sp.predict(X_sp)
+            prob_sp = model_sp.predict_proba(X_sp)
+
+        rtol = 2e-4
+        assert_allclose(pred, pred_sp, rtol=rtol)
+        assert_allclose(prob, prob_sp, rtol=rtol)
+        assert_allclose(model.coef_, model_sp.coef_, rtol=rtol)
+        assert_allclose(model.intercept_, model_sp.intercept_, rtol=rtol)

{scikit_learn_intelex-2024.1.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/manifold/t_sne.py

@@ -15,3 +15,7 @@
 # ===============================================================================
 
 from daal4py.sklearn.manifold import TSNE
+from onedal._device_offload import support_input_format
+
+TSNE.fit = support_input_format(queue_param=False)(TSNE.fit)
+TSNE.fit_transform = support_input_format(queue_param=False)(TSNE.fit_transform)

{scikit_learn_intelex-2024.1.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/metrics/pairwise.py

@@ -15,3 +15,8 @@
 # ===============================================================================
 
 from daal4py.sklearn.metrics import pairwise_distances
+from onedal._device_offload import support_input_format
+
+pairwise_distances = support_input_format(freefunc=True, queue_param=False)(
+    pairwise_distances
+)

{scikit_learn_intelex-2024.1.0.data → scikit_learn_intelex-2025.1.0.data}/data/Lib/site-packages/sklearnex/metrics/ranking.py

@@ -15,3 +15,6 @@
 # ===============================================================================
 
 from daal4py.sklearn.metrics import roc_auc_score
+from onedal._device_offload import support_input_format
+
+roc_auc_score = support_input_format(freefunc=True, queue_param=False)(roc_auc_score)