feat: return total number of iterations from algorithm

jolars · Dec 6, 2023 · e13ab6b · e13ab6b
1 parent a0d7f45
commit e13ab6b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/slope/fit_slope.h b/src/slope/fit_slope.h
@@ -113,6 +113,8 @@ fitSlope(const T& x,
 
   Clusters clusters(beta);
 
+  int it_total = 0;
+
   // Regularization path loop
   for (int path_step = 0; path_step < path_length; ++path_step) {
     if (params.print_level > 0) {
@@ -238,6 +240,7 @@ fitSlope(const T& x,
                             params);
         }
       }
+      it_total++;
     }
 
     // Store everything for this step of the path
@@ -259,7 +262,9 @@ fitSlope(const T& x,
   Eigen::SparseMatrix<double> betas(p, path_length);
   betas.setFromTriplets(beta_triplets.begin(), beta_triplets.end());
 
-  return { beta0s, betas, alpha, lambda, primals_path, dual_gaps_path };
+  return {
+    beta0s, betas, alpha, lambda, primals_path, dual_gaps_path, it_total
+  };
 }
 
 } // namespace slope
diff --git a/src/slope/results.h b/src/slope/results.h
@@ -13,6 +13,7 @@ struct Results
   const Eigen::ArrayXd lambda;
   const std::vector<std::vector<double>> primals;
   const std::vector<std::vector<double>> dual_gaps;
+  const int it_total;
 };
 
 } // namespace slope