scikit-learn GridSearchCV如何计算best_score_?

时间:2014-06-07 10:17:08

标签: python python-2.7 machine-learning scikit-learn

我一直在试图弄清楚GridSearchCV的best_score_参数是如何计算的(换句话说,它是什么意思)。 documentation说:

  

左侧数据的best_estimator得分。

所以,我试着把它翻译成我理解的东西并计算出实际“y”的r2_score和每个kfold的预测ys - 并得到不同的结果(使用这段代码):

test_pred = np.zeros(y.shape) * np.nan 
for train_ind, test_ind in kfold:
    clf.best_estimator_.fit(X[train_ind, :], y[train_ind])
    test_pred[test_ind] = clf.best_estimator_.predict(X[test_ind])
r2_test = r2_score(y, test_pred)

我到处寻找有关best_score_的更有意义的解释,但找不到任何东西。有人愿意解释吗?

由于

1 个答案:

答案 0 :(得分:9)

这是最佳估算器的平均交叉验证分数。让我们制作一些数据并修复交叉验证的数据划分。

>>> y = linspace(-5, 5, 200)
>>> X = (y + np.random.randn(200)).reshape(-1, 1)
>>> threefold = list(KFold(len(y)))

现在运行cross_val_scoreGridSearchCV,两者都有这些固定的折叠。

>>> cross_val_score(LinearRegression(), X, y, cv=threefold)
array([-0.86060164,  0.2035956 , -0.81309259])
>>> gs = GridSearchCV(LinearRegression(), {}, cv=threefold, verbose=3).fit(X, y) 
Fitting 3 folds for each of 1 candidates, totalling 3 fits
[CV]  ................................................................
[CV] ...................................... , score=-0.860602 -   0.0s
[Parallel(n_jobs=1)]: Done   1 jobs       | elapsed:    0.0s
[CV]  ................................................................
[CV] ....................................... , score=0.203596 -   0.0s
[CV]  ................................................................
[CV] ...................................... , score=-0.813093 -   0.0s
[Parallel(n_jobs=1)]: Done   3 out of   3 | elapsed:    0.0s finished

注意score=-0.860602输出中的score=0.203596score=-0.813093GridSearchCV;确切地说是cross_val_score返回的值。

请注意,“均值”实际上是折叠的宏观平均值。 iid GridSearchCV参数可用于获取样本的微观平均值。