【发布时间】:2020-04-18 11:11:08
【问题描述】:
假设我们使用 GridSearch 调整 SVM,如下所示:
algorithm = SVM()
parameters = {'kernel': ['rbf', 'sigmoid'], 'C': [0.1, 1, 10]}
grid= GridSearchCV(algorithm, parameters)
grid.fit(X, y)
然后您希望在cross_val_score 中使用最佳拟合参数/估计器。我的问题是,此时grid 是哪个型号?它是表现最好的吗?换句话说,我们可以这样做吗
cross_val_scores = cross_val_score(grid, X=X, y=y)
或者我们应该使用
cross_val_scores = cross_val_score(grid.best_estimator_, X=X, y=y)
当我同时运行两者时,我发现它们返回的分数不同,所以我很好奇这里的正确方法是什么。 (我假设使用best_estimator_。)不过,这引发了另一个问题,即:那么仅使用grid 作为模型有什么用?第一个?
【问题讨论】:
标签: python scikit-learn cross-validation k-fold