scikit-learn-intelex

scikit-learn-intelex

sklearn加速包

ref：

官方doc

概述：

Intel® Extension for Scikit-learn 是一种加速 Scikit-learn 应用程序的无缝方式。加速是通过使用英特尔® oneAPI 数据分析库 ( oneDAL ) 实现的。修补 scikit-learn 使其成为处理现实生活问题的非常适合的机器学习框架。

用法：

from sklearnex import patch_sklearn

patch_sklearn()

注意事项：

• sklearn和sklearn-intelex需要版本匹配，否则会导致程序虽然运行但是不执行，一直卡着

  • 示例版本：

    daal4py                           2021.2.2-py36ha9443f7_0 --> 2021.3.0-py36hae6d005_0
    dal                               2021.2.2-h06a4308_389 --> 2021.3.0-h06a4308_557
    scikit-learn-intex                2021.2.2-py36h06a4308_0 --> 2021.3.0-py36h06a4308_0
    scikit-learn              									0.24.2           py36ha9443f7_0