scikit-learn-intelex 2024.4.0__py312-none-win_amd64.whl → 2025.10.0__py312-none-win_amd64.whl

scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/__init__.py

@@ -0,0 +1,73 @@
+# ==============================================================================
+# Copyright 2014 Intel Corporation
+# Copyright 2024 Fujitsu Limited
+#
+# 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 platform
+
+if "Windows" in platform.system():
+    import os
+    import site
+    import sys
+
+    arch_dir = platform.machine()
+    plt_dict = {"x86_64": "intel64", "AMD64": "intel64", "aarch64": "arm"}
+    arch_dir = plt_dict[arch_dir] if arch_dir in plt_dict else arch_dir
+
+    current_path = os.path.dirname(__file__)
+    path_to_env = site.getsitepackages()[0]
+    path_to_libs = os.path.join(path_to_env, "Library", "bin")
+    if sys.version_info.minor >= 8:
+        if "DALROOT" in os.environ:
+            dal_root = os.environ["DALROOT"]
+            dal_root_redist = os.path.join(dal_root, "redist", arch_dir)
+            if os.path.exists(dal_root_redist):
+                os.add_dll_directory(dal_root_redist)
+                os.environ["PATH"] = dal_root_redist + os.pathsep + os.environ["PATH"]
+        if "TBBROOT" in os.environ:
+            tbb_root = os.environ["TBBROOT"]
+            tbb_root_redist = os.path.join(tbb_root, "bin")
+            if os.path.exists(tbb_root_redist):
+                os.add_dll_directory(tbb_root_redist)
+                os.environ["PATH"] = tbb_root_redist + os.pathsep + os.environ["PATH"]
+
+        try:
+            os.add_dll_directory(path_to_libs)
+        except FileNotFoundError:
+            pass
+
+    os.environ["PATH"] = path_to_libs + os.pathsep + os.environ["PATH"]
+
+try:
+    from daal4py._daal4py import *
+    from daal4py._daal4py import (
+        __has_dist__,
+        _get__daal_link_version__,
+        _get__daal_run_version__,
+        _get__version__,
+    )
+except ImportError as e:
+    s = str(e)
+    if "libfabric" in s:
+        raise ImportError(
+            s + "\n\nActivating your conda environment or sourcing mpivars."
+            "[c]sh/psxevars.[c]sh may solve the issue.\n"
+        )
+
+    raise
+
+from . import mb, sklearn
+
+__all__ = ["mb", "sklearn"]

scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/__main__.py

@@ -0,0 +1,58 @@
+# ==============================================================================
+# Copyright 2014 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 sys
+
+from .sklearn import patch_sklearn
+
+
+def _main():
+    import argparse
+
+    parser = argparse.ArgumentParser(
+        prog="python -m daal4py.sklearn_patches",
+        description="""
+            Run your Python script with Intel(R) Distribution for
+            Python* patches of scikit-learn, optimizing solvers of
+            scikit-learn with oneAPI Data Analytics Library.
+            """,
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument(
+        "-m", action="store_true", dest="module", help="Executes following as a module"
+    )
+    parser.add_argument("name", help="Script or module name")
+    parser.add_argument("args", nargs=argparse.REMAINDER, help="Command line arguments")
+    args = parser.parse_args()
+
+    try:
+        import sklearn
+
+        patch_sklearn()
+    except ImportError:
+        print("Scikit-learn could not be imported. Nothing to patch")
+
+    sys.argv = [args.name] + args.args
+    if "_" + args.name in globals():
+        return globals()["_" + args.name](*args.args)
+    import runpy
+
+    runf = runpy.run_module if args.module else runpy.run_path
+    runf(args.name, run_name="__main__")
+
+
+sys.exit(_main())

scikit_learn_intelex-2025.10.0.data/data/Lib/site-packages/daal4py/mb/__init__.py

@@ -0,0 +1,94 @@
+# Copyright contributors to the oneDAL project
+#
+# 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 sklearn.linear_model import LogisticRegression, SGDClassifier
+
+from .logistic_regression_builders import LogisticDAALModel
+from .tree_based_builders import GBTDAALBaseModel, GBTDAALModel
+
+__all__ = ["LogisticDAALModel", "GBTDAALModel", "convert_model"]
+
+
+def convert_model(model) -> "GBTDAALModel | LogisticDAALModel":
+    """
+    Convert GBT or LogReg models to Daal4Py
+
+    This function can be used to convert machine learning models / estimators
+    created through other libraries to daal4py classes which offer accelerated
+    prediction methods.
+
+    It supports gradient-boosted decision tree ensembles (GBT) from the libraries
+    ``xgboost``, ``lightgbm``, ``catboost``, and ``treelite``; and logistic regression
+    (binary and multinomial) models from scikit-learn.
+
+    See the documentation of the classes :obj:`daal4py.mb.GBTDAALModel` and
+    :obj:`daal4py.mb.LogisticDAALModel` for more details.
+
+    As an alternative to this function, models of a specific type (GBT or LogReg)
+    can also be instantiated by calling those classes directly - for example,
+    logistic regression models can be instantiated directly from fitted coefficients
+    and intercepts, thereby allowing to work with models from libraries beyond
+    scikit-learn.
+
+    Parameters
+    ----------
+    model : fitted model object
+        A fitted model object (either GBT or LogReg) from the supported libraries.
+
+    Returns
+    -------
+    obj : GBTDAALModel or LogisticDAALModel
+        A daal4py model object of the corresponding class for the model type, which
+        offers faster prediction methods.
+    """
+    if isinstance(model, LogisticRegression):
+        # The multi_class keyword is removed in scikit-learn 1.8, and OvR functionality
+        # has been replaced by other estimators. Therefore checking for linear classifiers
+        # only dependent on the solver.
+        if model.classes_.shape[0] > 2:
+            if not hasattr(model, "multi_class"):
+                if model.solver == "liblinear":
+                    raise TypeError(
+                        "Supplied 'model' object is a linear classifier, but not multinomial logistic"
+                    )
+            else:
+                if (model.multi_class == "ovr") or (
+                    model.multi_class == "auto" and model.solver == "liblinear"
+                ):
+                    raise TypeError(
+                        "Supplied 'model' object is a linear classifier, but not multinomial logistic"
+                        " (hint: pass multi_class='multinomial' to 'LogisticRegression')."
+                    )
+        elif (model.classes_.shape[0] == 2) and (
+            getattr(model, "multi_class", "auto") == "multinomial"
+        ):
+            raise TypeError(
+                "Supplied 'model' object is not a logistic regressor "
+                "(hint: pass multi_class='auto' to 'LogisticRegression')."
+            )
+        return LogisticDAALModel(model.coef_, model.intercept_)
+    if isinstance(model, SGDClassifier):
+        if model.classes_.shape[0] > 2:
+            raise TypeError(
+                "Supplied 'model' object is a linear classifier, but not multinomial logistic"
+                " (note: scikit-learn does not offer stochastic multinomial logistic models)."
+            )
+        if model.loss != "log_loss":
+            raise TypeError(
+                "Supplied 'model' object is not a logistic regressor "
+                "(hint: pass loss='log_loss' to 'SGDClassifier')."
+            )
+        return LogisticDAALModel(model.coef_, model.intercept_)
+
+    return GBTDAALModel(model)