gpt4 book ai didi

python - 如何加快cross_val_score的速度?

转载 作者:行者123 更新时间:2023-12-02 19:52:49 27 4
gpt4 key购买 nike

我想使用 sklearn.model_selection.cross_val_score 评估 MNIST 数据集上的 SGDClassifier。3折我花了大约6分钟。我怎样才能使用完整的系统功率来加速这个过程(我的意思是使用从CPU到显卡等的所有东西)顺便说一句,我正在监控 CPU 使用情况,它只使用了 54% 的电量。

from sklearn.datasets import fetch_openml
from sklearn.linear_model import SGDClassifier
from sklearn.model_selection import cross_val_score

mnist = fetch_openml('mnist_784')
X, y = mnist['data'], mnist['target']
X_train, X_test, y_train, y_test = X[:60000], X[60000:], y[:60000], y[60000:]
y_train_5 = (y_train == 5)
y_test_5 = (y_test == 5)

sgd_clf = SGDClassifier(random_state=42)
sgd_clf.fit(X_train, y_train)

cross_val_score(sgd_clf, X_train, y_train, cv=3, scoring='accuracy')

最佳答案

来自docs :

n_jobs : int or None, optional (default=None)

The number of CPUs to use to do the computation. None means 1 unless in a joblib.parallel_backend context. -1 means using all processors.

即您可以使用所有可用的核心

cross_val_score(sgd_clf, X_train, y_train, cv=3, scoring='accuracy', n_jobs=-1)

或者指定一些其他值n_jobs=k,如果使用所有核心会使您的机器变慢或无响应。

这将使用更多的 CPU 核心;据我所知,scikit-learn 中没有将计算卸载到 GPU 的功能。

关于python - 如何加快cross_val_score的速度?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57759943/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com