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 崩潰或結果無效,則重試 Trial 的最大次數。
  • max_consecutive_failed_trials:整數。默認為 3。連續失敗的 Trial 的最大數量。達到此數字後,搜索將停止。當所有重試均未成功時,Trial 將標記為失敗。