trying to fix cran error

ropensci · Jan 16, 2024 · 9d68083 · 9d68083
1 parent 45c5613
commit 9d68083
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/orsf_oop.cpp b/src/orsf_oop.cpp
@@ -537,7 +537,7 @@ double compute_mse_exported(arma::vec& y,
    // Load forest object if it was already grown
    if(!grow_mode){
 
-    uword n_obs = loaded_forest["n_obs"];
+    n_obs = loaded_forest["n_obs"];
 
     std::vector<uvec>                rows_oobag   = loaded_forest["rows_oobag"];
     std::vector<std::vector<double>> cutpoint     = loaded_forest["cutpoint"];
@@ -608,7 +608,7 @@ double compute_mse_exported(arma::vec& y,
 
     List forest_out;
     forest_out.push_back(n_obs, "n_obs");
-    forest_out.push_back(forest->get_oobag_denom(), "oobag_denom");
+    // forest_out.push_back(forest->get_oobag_denom(), "oobag_denom");
     forest_out.push_back(forest->get_rows_oobag(), "rows_oobag");
     forest_out.push_back(forest->get_cutpoint(), "cutpoint");
     forest_out.push_back(forest->get_child_left(), "child_left");
@@ -644,7 +644,13 @@ double compute_mse_exported(arma::vec& y,
     vec vi_output;
     if(run_forest){
      if(vi_type == VI_ANOVA){
-      vi_output = forest->get_vi_numer() / forest->get_vi_denom();
+
+      uvec denom = forest->get_vi_denom();
+      uvec zeros = find(denom == 0);
+      if(zeros.size() > 0) denom(zeros).fill(1);
+
+      vi_output = forest->get_vi_numer() / denom;
+
      } else {
       vi_output = forest->get_vi_numer() / n_tree;
      }