minSamplesLeaf 1

interpretml · Dec 23, 2024 · 2bd3277 · 2bd3277
1 parent ec42588
commit 2bd3277
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/shared/libebm/tests/boosting_unusual_inputs.cpp b/shared/libebm/tests/boosting_unusual_inputs.cpp
@@ -2128,7 +2128,7 @@ TEST_CASE("stress test, boosting") {
                const TermBoostFlags boostFlags = TermBoostFlags_Default;
 
                const double learningRate = 0.015625;
-               const IntEbm minSamplesLeaf = TestRand(rng, 5) + 1;
+               //const IntEbm minSamplesLeaf = TestRand(rng, 5) + 1;
                const double minHessian = 0 == TestRand(rng, 5) ? 0.015625 : 0.0;
                const double regAlpha = 0 == TestRand(rng, 5) ? 0.015625 : 0.0;
                const double regLambda = 0 == TestRand(rng, 5) ? 0.015625 : 0.0;
@@ -2147,7 +2147,7 @@ TEST_CASE("stress test, boosting") {
                validationMetricIteration = test.Boost(iTerm,
                                                      boostFlags,
                                                      learningRate,
-                                                     minSamplesLeaf,
+                                                     1,
                                                      minHessian,
                                                      regAlpha,
                                                      regLambda,
@@ -2165,5 +2165,5 @@ TEST_CASE("stress test, boosting") {
    }
 
    printf("\n%.17f\n", validationMetric);
-   CHECK(validationMetric == 11788.68786979941251047);
+   CHECK(validationMetric == 11791.24013081887096632);
 }