gpt4 book ai didi

machine-learning - sklearn.model_selection GridSearchCV 抛出 KeyError : 'mean_train_score'

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

我正在尝试使用 5 折交叉验证从 GridSearchCV 方法中的 classifierobject.cv_result_ 获取平均训练分数,但它给了我 *** KeyError: Google Colab 中的“mean_train_score”。但是,相同的代码在 sklearn 版本 0.19.1 的本地计算机 Ipython 笔记本中运行良好。谁能帮助我如何在 google colab 中获取输出?

clf.cv_results_.keys() 输出如下 -

对于我的本地笔记本 -

dict_keys(['mean_fit_time', 'std_fit_time', 'mean_score_time', 'std_score_time', 'param_n_neighbors', 'params', 'split0_test_score', 'split1_test_score', 'split2_test_score', 'split3_test_score', 'split4_test_score', 'mean_test_score', 'std_test_score', 'rank_test_score', 'split0_train_score', 'split1_train_score', 'split2_train_score', 'split3_train_score', 'split4_train_score', 'mean_train_score', 'std_train_score'])

在 Google Colab 笔记本上 -

dict_keys(['mean_fit_time', 'std_fit_time', 'mean_score_time', 'std_score_time', 'param_n_neighbors', 'params', 'split0_test_score', 'split1_test_score', 'split2_test_score', 'split3_test_score', 'split4_test_score', 'mean_test_score', 'std_test_score', 'rank_test_score'])

Google Colab 中的 mean_train_score 在哪里。我错过了什么吗?

from sklearn.model_selection import GridSearchCV

neigh = KNeighborsClassifier(n_jobs=-1)
parameters = {'n_neighbors':[1, 5, 10, 15, 19 , 21, 31, 41, 51]}
clf = GridSearchCV(neigh, parameters, cv=5,\
scoring='roc_auc',n_jobs=-1)
clf.fit(x_train, y_train)
train_auc= clf.cv_results_['mean_train_score']
train_auc_std= clf.cv_results_['std_train_score']
cv_auc = clf.cv_results_['mean_test_score']
cv_auc_std= clf.cv_results_['std_test_score']

最佳答案

尝试在 GridSearchCV() 中设置 return_train_score=True 来计算训练分数(默认情况下处于关闭状态,请参阅 docs )。也许您在 sklearn 中的某处本地设置中有某种全局变量。

关于machine-learning - sklearn.model_selection GridSearchCV 抛出 KeyError : 'mean_train_score' ,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57136676/

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