| mlr_pipeops_preproc_torch.augment_color_jitter {mlr3torch} | R Documentation |
PipeOpPreprocTorchAugmentColorJitter
Description
Calls torchvision::transform_color_jitter,
see there for more information on the parameters.
The preprocessing is applied row wise (no batch dimension).
Format
R6Class inheriting from PipeOpTaskPreprocTorch.
Parameters
| Id | Type | Default | Levels | Range |
| brightness | numeric | 0 | [0, \infty) |
|
| contrast | numeric | 0 | [0, \infty) |
|
| saturation | numeric | 0 | [0, \infty) |
|
| hue | numeric | 0 | [0, \infty) |
|
| stages | character | - | train, predict, both | - |
| affect_columns | untyped | selector_all() | - | |
[Package mlr3torch version 0.1.0 Index]