Merge pull request #54 from StochasticTree/serialization-hotfix

Fixed small BCF bug that shows up in serialization roundtrip
StochasticTree · Jun 20, 2024 · 42bba17 · 42bba17
2 parents d142a71 + dc68595
commit 42bba17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/bcf.R b/R/bcf.R
@@ -768,7 +768,7 @@ bcf <- function(X_train, Z_train, y_train, pi_train = NULL, group_ids_train = NU
         "b_leaf_tau" = b_leaf_tau,
         "outcome_mean" = y_bar_train,
         "outcome_scale" = y_std_train, 
-        "num_covariates" = ncol(X_train),
+        "num_covariates" = num_cov_orig,
         "num_prognostic_covariates" = sum(variable_weights_mu > 0),
         "num_treatment_covariates" = sum(variable_weights_tau > 0),
         "treatment_dim" = ncol(Z_train),