inst/doc/input_functions.R

In tfestimators: Interface to 'TensorFlow' Estimators

## ----setup, include=FALSE-----------------------------------------------------
# # https://www.tensorflow.org/get_started/input_fn
library(tfestimators)
knitr::opts_chunk$set(echo = TRUE)
knitr::opts_chunk$set(eval = FALSE)

## -----------------------------------------------------------------------------
#  model %>% train(
#    input_fn(mtcars,
#             features = c(drat, cyl),
#             response = mpg,
#             batch_size = 128,
#             epochs = 3)
#  )

## -----------------------------------------------------------------------------
#  model %>% train(
#    input_fn(mpg ~ drat + cyl,
#             data = mtcars,
#             batch_size = 128,
#             epochs = 3)
#  )

## -----------------------------------------------------------------------------
#  mtcars_input_fn <- function(data, ...) {
#    input_fn(data,
#             features = c("drat", "cyl"),
#             response = "mpg",
#             ...)
#  }

## -----------------------------------------------------------------------------
#  # train the model
#  model %>% train(mtcars_input_fn(train_data))
#  
#  # evaluate the model
#  model %>% evaluate(mtcars_input_fn(test_data))

## -----------------------------------------------------------------------------
#  m <- matrix(c(1:12), nrow = 4, ncol = 3)
#  colnames(m) <- c("x1", "x2", "y")
#  input_fn(m, features = c("x1", "x2"), response = "y")

## -----------------------------------------------------------------------------
#  input_fn(
#    object = list(
#      inputs = list(
#        list(list(1), list(2), list(3)),
#        list(list(4), list(5), list(6))),
#      output = list(
#        list(1, 2, 3), list(4, 5, 6))),
#    features = "inputs",
#    response = "output"
#  )