1、参数
网格搜索是sklearn.model_selection包下的GridSerchCV。
常用参数如下:
- estimator:分类器,需要传入需要寻优外的其他全部参数
- param_grid:分类器需要寻优的参数,是一个字典或者列表
- cv:交叉验证折数
2、代码
在鸢尾花数据集
上使用随机森林
,使用网格搜索
进行寻优
from sklearn.datasets import load_iris
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import GridSearchCV
# 加载数据集
iris = load_iris(<