如果我在scikit-learn库中使用GridSearchCV
来找到最佳模型,它将返回的最终模型是什么?也就是说,对于每组超参数,我们训练CV(比如3)模型的数量。这样,函数是否会返回这3个模型中的最佳模型以获得最佳参数设置?
答案 0 :(得分:2)
GridSearchCV
将返回一个包含大量信息的对象。它确实返回在剩余数据上表现最佳的模型:
best_estimator_:estimator或dict
搜索选择的估算器,即给出的估算器 左侧数据的最高分(或指定的最小损失)。 如果refit = False,则不可用。
请注意,这不是针对整个数据进行过培训的模型。这意味着,一旦您确信这是您想要的模型,您将需要自己重新训练整个数据模型。
参考:http://scikit-learn.org/stable/modules/generated/sklearn.model_selection.GridSearchCV.html