參數網格# (ParameterGrid)

class sklearn.model_selection.ParameterGrid(param_grid)[原始碼]#

參數網格，每個參數都有離散數值的集合。

可使用 Python 內建的 iter 函數來迭代參數值的組合。產生的參數組合順序是確定的。

更多資訊請參閱使用者指南。

參數:

param_grid字典 (key 為字串，value 為序列)，或此類字典的序列

要探索的參數網格，以字典形式將估算器的參數對應到允許值的序列。

空的字典表示使用預設參數。

字典的序列表示要搜尋的網格序列，這對於避免探索沒有意義或沒有效果的參數組合非常有用。請參閱下面的範例。

另請參閱

網格搜尋交叉驗證 (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