ParameterGrid#

class sklearn.model_selection.ParameterGrid(param_grid)[source]#

每个参数具有离散数量值的参数网格。

可用于使用 Python 内置函数 iter 迭代参数值组合。生成的参数组合的顺序是确定的。

欲了解更多信息，请阅读用户指南。

参数:

param_gridstr 到序列的字典，或此类序列

要探索的参数网格，作为将估计器参数映射到允许值序列的字典。

空字典表示默认参数。

字典序列表示要搜索的网格序列，可用于避免探索没有意义或没有效果的参数组合。请参阅下面的示例。

另请参阅

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

ParameterGrid#

本页