带有预定义拆分的Grid Search的Refit属性?

时间:2019-07-16 13:58:25

标签: scikit-learn gridsearchcv

我正在使用GridseachCV调整超参数。我分别有火车,验证数据。我正在使用预定义的拆分方法来处理此问题(类似于this):

ps = PredefinedSplit(test_fold=your_test_fold)

然后在cv=ps中设置GridSearchCV

如果我将改装参数设置为true,那么在所有参数调整之后,我的模型将在整个数据上训练还是仅在训练数据上训练?

1 个答案:

答案 0 :(得分:0)

来自the documentation

  

修饰:布尔值,字符串或可调用,默认为True

     

使用在整个数据集中找到的最佳参数来估算一个值。

因此,是的,它将使用具有最佳超参数的整个数据集。使用默认值(3折CV)还是来自PredefinedSplit的拆分都没关系。