Skip to content

Commit

Permalink
monotonicity
Browse files Browse the repository at this point in the history
  • Loading branch information
paulbkoch committed Dec 23, 2024
1 parent 6ca1cea commit 203e07f
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions shared/libebm/tests/boosting_unusual_inputs.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2140,9 +2140,9 @@ TEST_CASE("stress test, boosting") {
// we allow 1 cut more than the number of bins to test excessive leaves.
const IntEbm cLeaves = 1 + TestRand(rng, cRealBins + 1);
const std::vector<IntEbm> leaves(cDimensions, cLeaves);
const MonotoneDirection direction =
0 == TestRand(rng, 5) ? static_cast<MonotoneDirection>(TestRand(rng, 2) * 2 - 1) : 0;
const std::vector<MonotoneDirection> monotonicity(cDimensions, direction);
//const MonotoneDirection direction =
// 0 == TestRand(rng, 5) ? static_cast<MonotoneDirection>(TestRand(rng, 2) * 2 - 1) : 0;
//const std::vector<MonotoneDirection> monotonicity(cDimensions, direction);

validationMetricIteration = test.Boost(iTerm,
boostFlags,
Expand All @@ -2155,8 +2155,7 @@ TEST_CASE("stress test, boosting") {
categoricalSmoothing,
6,
0.75,
leaves,
monotonicity)
leaves)
.validationMetric;
}
}
Expand All @@ -2165,5 +2164,5 @@ TEST_CASE("stress test, boosting") {
}

printf("\n%.17f\n", validationMetric);
CHECK(validationMetric == 12195.17698180636580219);
CHECK(validationMetric == 12479.17282117725881108);
}

0 comments on commit 203e07f

Please sign in to comment.