document all functions

tidymodels · Dec 18, 2024 · 852c0d1 · 852c0d1
1 parent 82e93aa
commit 852c0d1
Show file tree

Hide file tree

Showing 5 changed files with 32 additions and 1 deletion.
diff --git a/R/model-partykit.R b/R/model-partykit.R
@@ -103,6 +103,9 @@ tidypredict_fit.party <- function(model) {
 }
 
 # For {orbital}
+#' Extract classprob trees for partykit models
+#' 
+#' For use in orbital package.
 #' @keywords internal
 #' @export
 .extract_partykit_classprob <- function(model) {

diff --git a/R/model-xgboost.R b/R/model-xgboost.R
@@ -185,6 +185,9 @@ tidypredict_fit.xgb.Booster <- function(model) {
 }
 
 # For {orbital}
+#' Extract processed xgboost trees
+#' 
+#' For use in orbital package.
 #' @keywords internal
 #' @export
 .extract_xgb_trees <- function(model) {

diff --git a/man/dot-extract_partykit_classprob.Rd b/man/dot-extract_partykit_classprob.Rd
diff --git a/man/dot-extract_xgb_trees.Rd b/man/dot-extract_xgb_trees.Rd
diff --git a/man/tidypredict-package.Rd b/man/tidypredict-package.Rd