GridSearchCV
是 scikit-learn 库中的一个工具,用于系统地遍历多种参数组合以找到最佳模型参数。它通过交叉验证来评估每个参数组合的性能,并选择表现最好的一组参数。
RandomizedSearchCV
来减少计算量。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
# 定义参数网格
param_grid = {'C': [0.1, 1, 10, 100], 'gamma': [1, 0.1, 0.01, 0.001], 'kernel': ['rbf', 'linear']}
# 创建模型
svc = SVC()
# 创建 GridSearchCV 对象
grid_search = GridSearchCV(svc, 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元无门槛券
手把手带您无忧上云