一款神器的 Python 工具库：2行代码百倍加速提升

最新推荐文章于 2024-09-22 15:50:52 发布

今天代码没bug

最新推荐文章于 2024-09-22 15:50:52 发布

阅读量231

点赞数 1

文章标签：大数据

本文链接：https://blog.csdn.net/m0_72091242/article/details/125442897

版权

本文介绍了sklearnex，一个可以显著提高Scikit-learn运算速度的工具库。借助英特尔® Extension for Scikit-learn，无需修改原有代码，即可实现10-100倍的效率提升。文章涵盖了sklearnex的安装、使用方法以及性能对比，展示了其在机器学习训练和推理中的巨大潜力。

摘要由CSDN通过智能技术生成

大家好，scikit-learn 作为经典的机器学习框架，其运算速度一直广受用户的诟病。

今天我给大家分享一个新工具包：sklearnex，可以在不改变原有代码的基础上，获得数十倍甚至上千倍的效率提升。

sklearnex简介

借助英特尔® Extension for Scikit-learn，您可以加速您的 Scikit-learn 应用程序，并且不需修改 Scikit-Learn原有代码结构。英特尔® Extension for Scikit-learn是一款免费软件 AI 加速器，可带来超过 10-100 倍的加速。

安装sklearnex

pip3 install notebook scikit-learn-intelex

快速上手

scikit-learn原代码

import numpy as np
from sklearn.cluster import KMeans

X = np.array([[1,  2], [1,  4], [1,  0],
              [10, 2], [10, 4], [10, 0]])

kmeans = KMeans(n_clusters=2, random_state=0).fit(X)
print(f"kmeans.labels_ = {kmeans.labels_}")

Run