enable everything except for purification

interpretml · Dec 23, 2024 · 11df118 · 11df118
1 parent 3793ebf
commit 11df118
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/shared/libebm/tests/boosting_unusual_inputs.cpp b/shared/libebm/tests/boosting_unusual_inputs.cpp
@@ -2089,18 +2089,20 @@ TEST_CASE("stress test, boosting") {
    }
    const size_t cRounds = 200;
    std::vector<IntEbm> boostFlagsAny{
-         TermBoostFlags_PurifyGain,
+         //TermBoostFlags_PurifyGain,
          TermBoostFlags_DisableNewtonGain,
          TermBoostFlags_DisableCategorical,
+         //TermBoostFlags_PurifyUpdate,
+         //TermBoostFlags_GradientSums, // does not return a metric
          TermBoostFlags_DisableNewtonUpdate,
          TermBoostFlags_RandomSplits
    };
    std::vector<IntEbm> boostFlagsChoose{
          TermBoostFlags_Default,
          TermBoostFlags_MissingLow,
          TermBoostFlags_MissingHigh,
-         //TermBoostFlags_MissingSeparate,
-         //TermBoostFlags_MissingDrop
+         TermBoostFlags_MissingSeparate,
+         TermBoostFlags_MissingDrop
    };
 
    double validationMetric = 0.0;
@@ -2167,5 +2169,5 @@ TEST_CASE("stress test, boosting") {
       }
    }
 
-   CHECK(validationMetric == 41977.238784308254);
+   CHECK(validationMetric == 41671.205484677885);
 }