R/torchts-model.R

In krzjoa/torchts: Time series Models with torch

#' Torchts abstract model
torchts_model <- function(class, net, index, key,
                          outcomes, predictors,
                          optim, timesteps,
                          parsed_formula,
                          horizon, device,
                          col_map_out,
                          extras){
  structure(
    class = c(class, "torchts_model"),
    list(
      net            = net,
      index          = index,
      key            = key,
      outcomes       = outcomes,
      predictors     = predictors,
      optim          = optim,
      timesteps      = timesteps,
      parsed_formula = parsed_formula,
      horizon        = horizon,
      device         = device,
      col_map_out    = col_map_out
    )
  )
}


#' @export
print.torchts_model <- function(x, ...){

  key <- if (length(x$key) == 0) "NULL" else x$key
  predictors <- paste0(x$predictors, collapse = ", ")
  outcomes   <- paste0(x$outcomes, collapse = ", ")

  print(x$net)
  cat("\n")
  cat("Model specification: \n")
  cli::cat_bullet(glue::glue("key: {key}"))
  cli::cat_bullet(glue::glue("index: {x$index}"))
  cli::cat_bullet(glue::glue("predictors: {predictors}"))
  cli::cat_bullet(glue::glue("outcomes: {outcomes}"))
  cli::cat_bullet(glue::glue("timesteps: {x$timesteps}"))
  cli::cat_bullet(glue::glue("horizon: {x$horizon}"))
  cli::cat_bullet(glue::glue("optimizer: {class(x$optim)[1]}"))

}