add model parameters to binary XBCF output for warm-start BCF initial…

…ization
JingyuHe · Feb 15, 2023 · afdecbf · afdecbf
1 parent 42e5ab6
commit afdecbf
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/R/XBCF.discrete.R b/R/XBCF.discrete.R
@@ -168,7 +168,14 @@ XBCF.discrete <- function(y, Z, X_con, X_mod, pihat = NULL, num_trees_con = 30,
     # store mean and sd in the model object (for predictions)
     obj$meany <- meany
     obj$sdy <- sdy
-
+
+    model_params <- list(n_sweeps = num_sweeps,
+                         n_burnin = burnin,
+                         n_trees_con = num_trees_con,
+                         n_trees_mod = num_trees_mod,
+                         pihat_status = "control")
+    obj$model_params <- model_params
+
     class(obj) <- "XBCFdiscrete"
     return(obj)
 }