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

大家好，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