diff --git a/src/regression.cpp b/src/regression.cpp index 90cd6d5..8bd1fdc 100644 --- a/src/regression.cpp +++ b/src/regression.cpp @@ -44,7 +44,7 @@ void check_beta_order(robin_hood::unordered_flat_map* beta, } float update_intercept(float *rowsum, float *Y, int n, float lambda, float intercept) { - float sumn = 0.0; + float sumn = intercept*n; for (int i = 0; i < n; i++) { sumn -= rowsum[i]; }