ray.tune.grid_search

ray.tune.grid_search(values: Iterable) → Dict[str, Iterable]

指定一个值的网格以进行搜索。

在网格搜索中指定的值保证会被采样。

如果定义了多个网格搜索变量，它们将通过组合产品进行组合。这意味着将采样每个可能的值组合。

示例

>>> from ray import tune
>>> param_space={
...   "x": tune.grid_search([10, 20]),
...   "y": tune.grid_search(["a", "b", "c"])
... }

这将创建一个包含6个样本的网格：{"x": 10, "y": "a"}, {"x": 10, "y": "b"}, 等等。

在 TuneConfig 中指定 num_samples 时，这将指定每个网格搜索组合的随机样本数量。

例如，在上面的例子中，如果 num_samples=4，总共将启动 24 次试验 - 每个网格搜索组合有 4 次试验，共 6 个组合。

参数:

values – 一个可迭代对象，其参数将用于创建试验网格。

PublicAPI (测试版): 此API目前处于测试阶段，在成为稳定版本之前可能会发生变化。