Parameters for residual networks — resnet-param • dials

These parameters are used with tabular residual networks, such as tabular_resnet() when fit with the brulee engine.

Usage

bottleneck_units(range = c(2L, 25L), trans = NULL)

resid_at(range = c(2L, unknown()), trans = NULL)

Arguments

range: A two-element vector holding the defaults for the smallest and largest possible values, respectively. If a transformation is specified, these values should be in the transformed units.
trans: A trans object from the scales package, such as scales::transform_log10() or scales::transform_reciprocal(). If not provided, the default is used which matches the units used in range. If no transformation, NULL.

Details

bottleneck_units(): The number of units used in ResNet layers within a residual block.
resid_at(): The layer indices at which residual connections are added. The upper bound depends on the number of hidden layers in the network and so is left as unknown() by default.

Examples

bottleneck_units()
#> # Bottleneck Units (quantitative)
#> Range: [2, 25]
resid_at()
#> Residual Locations (quantitative)
#> Range: [2, ?]