Keras 3 API 文档 / KerasTuner / 预言机 / 网格搜索预言机

网格搜索预言机

[来源]

GridSearchOracle

keras_tuner.oracles.GridSearchOracle(
    objective=None,
    max_trials=None,
    seed=None,
    hyperparameters=None,
    allow_new_entries=True,
    tune_new_entries=True,
    max_retries_per_trial=0,
    max_consecutive_failed_trials=3,
)

网格搜索预言机。

参数

  • objective: 字符串、keras_tuner.Objective 实例或 keras_tuner.Objective 和字符串的列表。如果为字符串,则会推断优化方向(最小化或最大化)。如果为 keras_tuner.Objective 列表,我们将最小化所有目标的总和以最小化,减去所有目标的总和以最大化。当 Tuner.run_trial()HyperModel.fit() 返回单个浮点数作为要最小化的目标时,objective 参数是可选的。
  • max_trials: 可选整数,最多测试的试验(模型配置)总数。请注意,如果搜索空间已用尽,预言机可能会在测试 max_trial 个模型之前中断搜索。如果未指定,它将一直运行到搜索空间用尽为止。
  • seed: 可选整数,随机种子。
  • hyperparameters: 可选 HyperParameters 实例。可用于覆盖(或预先注册)搜索空间中的超参数。
  • tune_new_entries: 布尔值,超模型请求但未在 hyperparameters 中指定的超参数条目是否应添加到搜索空间中,还是不应添加到搜索空间中。如果不是,则将使用这些参数的默认值。默认为 True。
  • allow_new_entries: 布尔值,超模型是否允许请求 hyperparameters 中未列出的超参数条目。默认为 True。
  • max_retries_per_trial: 整数。默认为 0。如果试验崩溃或结果无效,重试 Trial 的最大次数。
  • max_consecutive_failed_trials: 整数。默认为 3。连续失败的 Trial 的最大数量。达到此数量后,搜索将停止。当所有重试均未成功时,Trial 将标记为失败。