limit number of trees in example

ModelOriented · Sep 28, 2023 · e5147cd · e5147cd
1 parent 31cec37
commit e5147cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/R/unify_xgboost.R b/R/unify_xgboost.R
@@ -26,7 +26,7 @@
 #' target <- fifa20$target
 #' param <- list(objective = "reg:squarederror", max_depth = 3)
 #' xgb_model <- xgboost::xgboost(as.matrix(data), params = param, label = target,
-#'                               nrounds = 200, verbose = 0)
+#'                               nrounds = 20, verbose = 0)
 #' unified_model <- xgboost.unify(xgb_model, as.matrix(data))
 #' shaps <- treeshap(unified_model, data[1:2,])
 #' plot_contribution(shaps, obs = 1)

diff --git a/man/xgboost.unify.Rd b/man/xgboost.unify.Rd