tests/testthat/test_keras.R

skip_on_os("windows")
skip_if_not_installed("keras")

test_that("get_keras_optimizer", {
  expect_is(get_keras_optimizer("adadelta"), "keras.optimizer_v2.adadelta.Adadelta")
  expect_is(get_keras_optimizer("adagrad"), "keras.optimizer_v2.adagrad.Adagrad")
  expect_is(get_keras_optimizer("adamax"), "keras.optimizer_v2.adamax.Adamax")
  expect_is(get_keras_optimizer("adam"), "keras.optimizer_v2.adam.Adam")
  expect_is(get_keras_optimizer("nadam"), "keras.optimizer_v2.nadam.Nadam")
  expect_is(get_keras_optimizer("rmsprop"), "keras.optimizer_v2.rmsprop.RMSprop")
  expect_is(get_keras_optimizer("sgd"), "keras.optimizer_v2.gradient_descent.SGD")
})

Try the survivalmodels package in your browser

Any scripts or data that you put into this service are public.

survivalmodels documentation built on March 24, 2022, 9:05 a.m.