document all functions

Author: EmilHvitfeldt

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) {

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) {