GridSearchCV是一种参数自动搜索的方法,主要用于模型选择和调参。它通过遍历给定的参数组合,对每个组合进行交叉验证(Cross-Validation,简称CV),并选择表现最好的参数组合。这种方法可以帮助我们在有限的计算资源下,找到最优的模型参数。
以下是一个使用GridSearchCV进行参数搜索的示例代码:
from sklearn.model_selection import GridSearchCV
from sklearn.svm import SVC
from sklearn.datasets import load_iris
# 加载数据集
iris = load_iris()
X = iris.data
y = iris.target
# 定义模型
model = SVC()
# 定义参数网格
param_grid = {
'C': [0.1, 1, 10],
'kernel': ['linear', 'rbf'],
'gamma': ['scale', 'auto']
}
# 创建GridSearchCV对象
grid_search = GridSearchCV(estimator=model, param_grid=param_grid, cv=5)
# 进行网格搜索
grid_search.fit(X, y)
# 输出最佳参数
print("Best parameters found: ", grid_search.best_params_)
print("Best cross-validation score: ", grid_search.best_score_)
通过以上内容,你应该对GridSearchCV及其相关概念有了全面的了解,并能解决一些常见问题。
领取专属 10元无门槛券
手把手带您无忧上云