GridSearchCV最终模型

时间:2018-03-23 17:59:17

标签: python machine-learning scikit-learn

如果我在scikit-learn库中使用GridSearchCV来找到最佳模型,它将返回的最终模型是什么?也就是说,对于每组超参数,我们训练CV(比如3)模型的数量。这样,函数是否会返回这3个模型中的最佳模型以获得最佳参数设置?

1 个答案:

答案 0 :(得分:2)

GridSearchCV将返回一个包含大量信息的对象。它确实返回在剩余数据上表现最佳的模型:

  

best_estimator_:estimator或dict

     

搜索选择的估算器,即给出的估算器   左侧数据的最高分(或指定的最小损失)。   如果refit = False,则不可用。

请注意,这不是针对整个数据进行过培训的模型。这意味着,一旦您确信这是您想要的模型,您将需要自己重新训练整个数据模型。

参考:http://scikit-learn.org/stable/modules/generated/sklearn.model_selection.GridSearchCV.html