add catch for failing code

interpretml · Jan 26, 2025 · 5146aa5 · 5146aa5
1 parent e87e320
commit 5146aa5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/shared/libebm/PartitionOneDimensionalBoosting.cpp b/shared/libebm/PartitionOneDimensionalBoosting.cpp
@@ -1088,6 +1088,11 @@ template<bool bHessian, size_t cCompilerScores> class PartitionOneDimensionalBoo
             EBM_ASSERT(std::isnan(gainParent) || 0 <= gainParent);
 
             const FloatCalc gain = gainMissing + gainNonMissing - gainParent;
+            if (gain < 0.0) {
+               printf("GARBAGAGA: %e, %e, %e, %e\n", gain, gainMissing, gainNonMissing, gainParent);
+               exit(1);
+            }
+
             totalGain += gain;
 
             ++iScore;