diff --git a/R/hardhat.R b/R/hardhat.R index 5f68d5c..be2aaf6 100644 --- a/R/hardhat.R +++ b/R/hardhat.R @@ -246,7 +246,7 @@ new_tabnet_fit <- function(fit, blueprint) { #' * A __matrix__ of predictors. #' * A __recipe__ specifying a set of preprocessing steps #' created from [recipes::recipe()]. -#' * A __Node__ where tree will be used as hierarchical outcome, +#' * A __Node__ where tree leaves will be left out, #' and attributes will be used as predictors. #' #' The predictor data should be standardized (e.g. centered or scaled). diff --git a/man/tabnet_pretrain.Rd b/man/tabnet_pretrain.Rd index 0d85757..7a861f8 100644 --- a/man/tabnet_pretrain.Rd +++ b/man/tabnet_pretrain.Rd @@ -55,7 +55,7 @@ tabnet_pretrain(x, ...) \item A \strong{matrix} of predictors. \item A \strong{recipe} specifying a set of preprocessing steps created from \code{\link[recipes:recipe]{recipes::recipe()}}. -\item A \strong{Node} where tree will be used as hierarchical outcome, +\item A \strong{Node} where tree leaves will be left out, and attributes will be used as predictors. }