From e8f5d3009b2ff2869bc758dd213d80d5121a2386 Mon Sep 17 00:00:00 2001 From: "C. Regouby" Date: Sat, 20 Jul 2024 15:08:17 +0200 Subject: [PATCH] cleared Node pretrain help --- R/hardhat.R | 2 +- man/tabnet_pretrain.Rd | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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. }