gpt4 book ai didi

python - 如何为 ML 模型正确设置种子值?

转载 作者:行者123 更新时间:2023-11-30 09:30:40 27 4
gpt4 key购买 nike

我是机器学习新手,正在运行不同的分类模型。我观察到每次运行模型时,都会得到略有不同的结果。我在网上了解到这是关于设置种子值。但我无法实现可重复性?

下面是我尝试设置种子值的代码,但没有帮助

from sklearn.svm import SVC
random.seed(1234)
param_grid = {'C': [0.001,0.01,0.1,1,10,100],
'gamma': [1,0.1,0.01,0.001],
'kernel': ['linear','rbf','poly'],
'class_weight':['balanced']}
svm=SVC()
svm_cv=GridSearchCV(svm,param_grid,cv=5)
svm_cv.fit(X_train_std,y_train)
y_pred = svm_cv.predict(X_test_std)
cm = confusion_matrix(y_test, y_pred)
print(cm)
print("Accuracy is ", accuracy_score(y_test, y_pred))

您能否帮助我了解如何设置 seed 值,以便每次运行上述代码时,我都能得到相同的结果/准确性/指标

最佳答案

Scikit learn 使用 numpy 随机种子。因此,您应该导入 numpy 并设置其随机种子,如下所示:

import numpy as np
np.random.seed(1234)

( https://www.mikulskibartosz.name/how-to-set-the-global-random_state-in-scikit-learn/ )

关于python - 如何为 ML 模型正确设置种子值?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59372994/

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