Skip to content

Commit

Permalink
remove boost flags
Browse files Browse the repository at this point in the history
  • Loading branch information
paulbkoch committed Dec 26, 2024
1 parent 9030396 commit 14aba7a
Showing 1 changed file with 15 additions and 11 deletions.
26 changes: 15 additions & 11 deletions shared/libebm/tests/boosting_unusual_inputs.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2090,18 +2090,22 @@ static double RandomizedTesting(const AccelerationFlags acceleration) {
// // terms.push_back({0, 1, 2, 3}); // TODO: enable when fast enough
//}

std::vector<IntEbm> boostFlagsAny{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,
std::vector<IntEbm> boostFlagsAny{
//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_MissingHigh,
//TermBoostFlags_MissingSeparate,
//TermBoostFlags_MissingDrop
};

double validationMetric = 1.0;
for(IntEbm classesCount = Task_Regression; classesCount < 5; ++classesCount) {
Expand Down

0 comments on commit 14aba7a

Please sign in to comment.