R/aaa.R

In brulee: High-Level Modeling Functions with 'torch'

#' @import torch
#' @import rlang
#' @importFrom stats complete.cases model.matrix terms
#' @importFrom utils globalVariables
#'
#' @importFrom dplyr %>%
#' @export
dplyr::`%>%`

#' @importFrom ggplot2 autoplot
#' @export
ggplot2::autoplot

#' @importFrom generics tunable
#' @export
generics::tunable

#' @importFrom stats coef
#' @export
stats::coef

# ------------------------------------------------------------------------------

utils::globalVariables(
 c(
  "object", "iteration", "loss"
 )
)

# ------------------------------------------------------------------------------

# nocov start
.onAttach <- function(libname, pkgname) {
 s3_register("ggplot2::autoplot", "brulee_mlp")
 invisible()
}

# Dynamic reg helper -----------------------------------------------------------

# vctrs/register-s3.R
# https://github.com/r-lib/vctrs/blob/master/R/register-s3.R
s3_register <- function(generic, class, method = NULL) {
 stopifnot(is.character(generic), length(generic) == 1)
 stopifnot(is.character(class), length(class) == 1)

 pieces <- strsplit(generic, "::")[[1]]
 stopifnot(length(pieces) == 2)
 package <- pieces[[1]]
 generic <- pieces[[2]]

 if (is.null(method)) {
  method <- get(paste0(generic, ".", class), envir = parent.frame())
 }
 stopifnot(is.function(method))

 if (package %in% loadedNamespaces()) {
  registerS3method(generic, class, method, envir = asNamespace(package))
 }

 # Always register hook in case package is later unloaded & reloaded
 setHook(
  packageEvent(package, "onLoad"),
  function(...) {
   registerS3method(generic, class, method, envir = asNamespace(package))
  }
 )
}

# nocov end