tests/testthat/test-dataset-fitting.R

In rstudio/tfdatasets: Interface to 'TensorFlow' Datasets

context("dataset-fitting")


library(keras3)

test_succeeds("can fit a dataset using keras fit", {
  d  <-
    tensor_slices_dataset(list(matrix(1:20, ncol = 2), matrix(1:10, ncol = 1))) %>%
    dataset_batch(2) %>%
    dataset_shuffle(1024)

  skip_if(tf_version() < "2.16")
  model <-
    keras_model_sequential(input_shape = 2) %>%
    layer_dense(units = 1)

  model %>% compile(optimizer = "adam", loss = "mse")
  model %>% fit(d, steps_per_epoch = 5, epochs = 1)
})