ParameterGrid#
- class sklearn.model_selection.ParameterGrid(param_grid)[source]#
每个参数具有离散数值的网格。
可用于使用Python内置的迭代器函数(iter)遍历参数值组合。生成的参数组合的顺序是确定的。
在用户指南中阅读更多内容。
- 参数:
- param_grid字典(str到序列),或此类序列
要探索的参数网格,表示为一个字典,将估计器参数映射到允许值的序列。
空字典表示默认参数。
字典序列表示要搜索的网格序列,可用于避免探索无意义或无效果的参数组合。请参阅下面的示例。
另请参阅
GridSearchCV
使用
ParameterGrid
执行完整的并行化参数搜索。
示例
>>> from sklearn.model_selection import ParameterGrid >>> param_grid = {'a': [1, 2], 'b': [True, False]} >>> list(ParameterGrid(param_grid)) == ( ... [{'a': 1, 'b': True}, {'a': 1, 'b': False}, ... {'a': 2, 'b': True}, {'a': 2, 'b': False}]) True
>>> grid = [{'kernel': ['linear']}, {'kernel': ['rbf'], 'gamma': [1, 10]}] >>> list(ParameterGrid(grid)) == [{'kernel': 'linear'}, ... {'kernel': 'rbf', 'gamma': 1}, ... {'kernel': 'rbf', 'gamma': 10}] True >>> ParameterGrid(grid)[1] == {'kernel': 'rbf', 'gamma': 1} True