Skip to content

Commit 56242bb

Browse files
calculating discretized beta using sample_weight_discretized
1 parent 6a3f26d commit 56242bb

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

cpp/term.h

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -532,16 +532,8 @@ void Term::calculate_coefficient_and_error_on_discretized_data(bool direction_ri
532532
double xwy{0};
533533
for (size_t i = index_start; i <= index_end; ++i)
534534
{
535-
if(sample_weight_discretized.size()>0)
536-
{
537-
xwx+=values_sorted[i]*values_sorted[i]*sample_weight_discretized[i];
538-
xwy+=values_sorted[i]*y_discretized[i]*sample_weight_discretized[i];
539-
}
540-
else
541-
{
542-
xwx+=values_sorted[i]*values_sorted[i];
543-
xwy+=values_sorted[i]*y_discretized[i];
544-
}
535+
xwx+=values_sorted[i]*values_sorted[i]*sample_weight_discretized[i];
536+
xwy+=values_sorted[i]*y_discretized[i]*sample_weight_discretized[i];
545537
}
546538
if(xwx!=0)
547539
{

0 commit comments

Comments
 (0)