Update piecewise_linear_model.hpp

gvinciguerra · Nov 1, 2024 · 5d25a6b · 5d25a6b
1 parent 77f441c
commit 5d25a6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/pgm/piecewise_linear_model.hpp b/include/pgm/piecewise_linear_model.hpp
@@ -303,7 +303,7 @@ size_t make_segmentation(size_t n, size_t start, size_t end, size_t epsilon, Fin
             add_point(in(i), i);
         }
     }
-    if (end >= 2 && in(end - 1) != in(end - 2))
+    if (end >= start + 2 && in(end - 1) != in(end - 2))
         add_point(in(end - 1), end - 1);
 
     if (end == n) {