是一种优化超参数的方法。超参数是在机器学习算法中需要手动设置的参数,它们决定了模型的性能和行为。网格搜索是一种穷举搜索的方法,通过遍历给定的超参数组合,找到最佳的超参数组合,以获得最优的模型性能。
具体步骤如下:
from sklearn.model_selection import GridSearchCV
from sklearn.svm import SVC
X = ...
y = ...
model = SVC()
param_grid = {
'C': [0.1, 1, 10],
'kernel': ['linear', 'rbf'],
'gamma': [0.1, 1, 10],
'degree': [2, 3, 4],
'coef0': [0.0, 0.5, 1.0],
'shrinking': [True, False],
'probability': [True, False],
'tol': [0.001, 0.01, 0.1]
}
grid_search = GridSearchCV(estimator=model, param_grid=param_grid, cv=5)
grid_search.fit(X, y)
best_params = grid_search.best_params_
best_score = grid_search.best_score_
print("Best Parameters: ", best_params)
print("Best Score: ", best_score)
网格搜索通过遍历所有可能的超参数组合,对每个组合进行交叉验证,评估模型性能,并找到最佳的超参数组合。这样可以提高模型的准确性和泛化能力。
推荐的腾讯云相关产品:腾讯云机器学习平台(Tencent Machine Learning Platform,TMLP) 产品介绍链接地址:https://cloud.tencent.com/product/tmlp
领取专属 10元无门槛券
手把手带您无忧上云