get_optimizer: Get Keras Optimizer

In mlr3learners/mlr3learners.proba: Learners for mlr3proba

Description

Utility function to construct optimiser from keras, primarily for internal use.

Usage

get_optimizer(
  optimizer = "adam",
  lr = 0.02,
  beta_1 = 0.9,
  beta_2 = 0.999,
  epsilon = NULL,
  decay = 0,
  clipnorm = NULL,
  clipvalue = NULL,
  schedule_decay = 0.004,
  momentum = 0,
  nesterov = FALSE
)

Arguments

optimizer	(character(1)) Optimizer to construct, see details for those available. Default is "adam".
lr	(numeric(1)) Passed to all optimizers except adadelta and adagrad.
beta_1, beta_2, epsilon	(numeric(1)) Passed to adamax, adam, and nadam.
decay	(numeric(1)) Passed to adamax, adam, and sgd.
clipnorm, clipvalue	(numeric(1)) Passed to adamax, adam, nadam, and sgd.
schedule_decay	(numeric(1)) Passed to nadam.
momentum	(numeric(1)) Passed to sgd.
nesterov	(logical(1)) Passed to sgd.

Details

Implement optimizers are

• "adadelta"
  keras::optimizer_adadelta

• "adagrad"
  keras::optimizer_adagrad

• "adamax"
  keras::optimizer_adamax

• "adam"
  keras::optimizer_adam

• "nadam"
  keras::optimizer_nadam

• "rmsprop"
  keras::optimizer_rmsprop

• "sgd"
  keras::optimizer_sgd

mlr3learners/mlr3learners.proba documentation built on Aug. 9, 2020, 12:50 a.m.