Python sklearn ParameterGrid用法及代碼示例

本文簡要介紹python語言中 sklearn.model_selection.ParameterGrid 的用法。

用法:

class sklearn.model_selection.ParameterGrid(param_grid)

參數網格，每個參數具有離散數量的值。

可用於使用 Python 內置函數 iter 迭代參數值組合。生成的參數組合的順序是確定的。

在用戶指南中閱讀更多信息。

參數：

param_grid：str 的字典到序列，或這樣的序列

要探索的參數網格，作為將估計器參數映射到允許值序列的字典。

空字典表示默認參數。

dicts 序列表示要搜索的網格序列，並且對於避免探索沒有意義或沒有效果的參數組合很有用。請參閱下麵的示例。

例子：

>>> 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