tests/testthat/helper-keras.R

In nn2poly: Neural Network Weights Transformation into Polynomial Coefficients

keras_test_model <- function() {
  tensorflow::set_random_seed(42)

  nn <- keras::keras_model_sequential()
  nn <- keras::layer_dense(nn, units = 2, activation = "tanh", input_shape = 2)
  nn <- keras::layer_dense(nn, units = 3, activation = "softplus")
  nn <- keras::layer_dense(nn, units = 2, activation = "linear")

  nn
}