iterable_dataset function

Creates an iterable dataset

iterable_dataset(
  name,
  inherit = IterableDataset,
  ...,
  private = NULL,
  active = NULL,
  parent_env = parent.frame()
)

Arguments

• name: a name for the dataset. It it's also used as the class for it.
• inherit: you can optionally inherit from a dataset when creating a new dataset.
• ...: public methods for the dataset class
• private: passed to R6::R6Class().
• active: passed to R6::R6Class().
• parent_env: An environment to use as the parent of newly-created objects.

Examples

if (torch_is_installed()) {
ids <- iterable_dataset(
  name = "hello",
  initialize = function(n = 5) {
    self$n <- n
    self$i <- 0
  },
  .iter = function() {
    i <- 0
    function() {
      i <<- i + 1
      if (i > self$n) {
        coro::exhausted()
      } else {
        i
      }
    }
  }
)
coro::collect(ids()$.iter())
}