feat: return alpha and lambda from slope()

jolars · Dec 5, 2023 · d3294ed · d3294ed
1 parent 6bdeeda
commit d3294ed
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/slope/results.h b/src/slope/results.h
@@ -9,6 +9,8 @@ struct Results
 {
   const Eigen::VectorXd beta0s;
   const Eigen::SparseMatrix<double> betas;
+  const Eigen::ArrayXd alpha;
+  const Eigen::ArrayXd lambda;
   const std::vector<double> primals;
 };
 

diff --git a/src/slope/slope.h b/src/slope/slope.h
@@ -298,7 +298,7 @@ slope(const T& x,
   Eigen::SparseMatrix<double> betas(p, path_length);
   betas.setFromTriplets(beta_triplets.begin(), beta_triplets.end());
 
-  return { beta0s, betas, primals };
+  return { beta0s, betas, alpha, lambda, primals };
 }
 
 } // namespace slope