mlr3 超频带调谐中的配置数

Question

如何控制在 mlr3 的超频带调谐期间评估的配置数量？我注意到当我在 xgboost() 中调整 6 个参数时，代码评估了大约 9 个配置。当我在 catboost() 中调整相同数量的参数时，代码从评估 729 个配置开始。我在这两种情况下都使用 eta = 3。

Answer 1

超频带中采样配置的数量由预算超参数和 eta 的下限和上限定义。您可以预览时间表和配置数量：

mlr3hyperband::hyperband_schedule(r_min = 1, r_max = 81, eta = 3)

#>     bracket stage budget  n
#>  1:       4     0      1 81
#>  2:       4     1      3 27
#>  3:       4     2      9  9
#>  4:       4     3     27  3
#>  5:       4     4     81  1
#>  6:       3     0      3 34
#>  7:       3     1      9 11
#>  8:       3     2     27  3
#>  9:       3     3     81  1
#> 10:       2     0      9 15
#> 11:       2     1     27  5
#> 12:       2     2     81  1
#> 13:       1     0     27  8
#> 14:       1     1     81  2
#> 15:       0     0     81  5

mlr3hyperband::hyperband_n_configs(r_min = 1, r_max = 81, eta = 3)

#> 143

如果要评估更多配置，请增加 hyperband 的 repetition 参数。 Hyperband 将在评估最后一个括号后重新开始。为此 devtools::install_github("mlr-org/mlr3hyperband").

安装最新的 gh 版本

mlr3 超频带调谐中的配置数

Number of configurations in mlr3 hyperband tuning

mlr3