R/autoencoder_weight_decay.R
add_weight_decay.Rd
Adds a weight decay regularization to the encoding layer of a given autoencoder
add_weight_decay(learner, decay = 0.02)
learner | The |
---|---|
decay | Numeric value indicating the amount of decay |
An autoencoder object which contains the weight decay