ParameterGrid

classsklearn.model_selection.ParameterGrid(param_grid)[исходный код]

Сетка параметров, каждый из которых принимает конечное число значений.

Можно использовать для перебора комбинаций значений параметров с помощью встроенной функции Python iter. Порядок сгенерированных комбинаций параметров детерминирован.

Подробнее см. в руководстве пользователя.

Параметры:

param_gridсловарь: строки — последовательности, или последовательность таких словарей

Сетка параметров для исследования в виде словаря, сопоставляющего параметры оценивателя с последовательностями допустимых значений.

Пустой словарь означает использование параметров по умолчанию.

Последовательность словарей задаёт последовательность сеток для поиска и полезна, чтобы не исследовать комбинации параметров, которые не имеют смысла или не влияют на результат. См. примеры ниже.

См. также

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