#' @import torch
#' @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
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.