Trying to fix the bug #191 again

config-i1 · Feb 7, 2022 · 428cd7d · 428cd7d
1 parent 99eb910
commit 428cd7d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/ssGeneral.cpp b/src/ssGeneral.cpp
@@ -884,7 +884,8 @@ List backfitter(arma::mat &matrixVt, arma::mat const &matrixF, arma::rowvec cons
 
             // This is a failsafe for cases of ridiculously high and ridiculously low values
             if(vecYfit(i-lagsModelMax) > 1e+100){
-                if(i-lagsModelMax==0){
+                // If this is the first value, substitute with zero
+                if(i==lagsModelMax){
                     vecYfit(i-lagsModelMax) = 0;
                 }
                 else{
@@ -982,7 +983,8 @@ List backfitter(arma::mat &matrixVt, arma::mat const &matrixF, arma::rowvec cons
 
             // This is for cases of ridiculously high and ridiculously low values
             if(vecYfit(i-lagsModelMax) > 1e+100){
-                if(i-lagsModelMax==obs+lagsModelMax){
+                // If this is the first value, substitute with zero
+                if(i==obs+lagsModelMax-1){
                     vecYfit(i-lagsModelMax) = 0;
                 }
                 else{