From 946e522cc477d5c17098654b1a668426db586489 Mon Sep 17 00:00:00 2001 From: Mathieu Guillame-Bert Date: Mon, 28 Oct 2024 04:12:58 -0700 Subject: [PATCH] Disable non-interresting & slow experiments from the training benchmark. PiperOrigin-RevId: 690554064 --- .../cli/monitoring/benchmark_training.cc | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/yggdrasil_decision_forests/cli/monitoring/benchmark_training.cc b/yggdrasil_decision_forests/cli/monitoring/benchmark_training.cc index 7d800f83..00bd2bad 100644 --- a/yggdrasil_decision_forests/cli/monitoring/benchmark_training.cc +++ b/yggdrasil_decision_forests/cli/monitoring/benchmark_training.cc @@ -345,6 +345,7 @@ GBTVariations() { // Make sure PRESORTED is always used. // Note: Ideally, FORCE_PRESORTED should be renamed to PRESORTED. + /* {" force_presorted", [](model::gradient_boosted_trees::proto:: GradientBoostedTreesTrainingConfig* gbt_config) { @@ -354,6 +355,7 @@ GBTVariations() { model::decision_tree::proto::DecisionTreeTrainingConfig:: Internal::FORCE_PRESORTED); }}, + */ }; } @@ -678,12 +680,13 @@ absl::Status Benchmark_GBT_Synthetic(std::vector* results) { .num_features = 200, .resolution = 1000, .use_discretized_numerical_features = true}, - {.num_examples = 4'000'000, .num_features = 200}, - {.num_examples = 4'000'000, .num_features = 200, .resolution = 1000}, - {.num_examples = 4'000'000, - .num_features = 200, - .resolution = 1000, - .use_discretized_numerical_features = true}, + // {.num_examples = 4'000'000, .num_features = 200}, + // {.num_examples = 4'000'000, .num_features = 200, .resolution = + // 1000}, + // {.num_examples = 4'000'000, + // .num_features = 200, + // .resolution = 1000, + // .use_discretized_numerical_features = true}, }) { for (const auto& strategy : GBTVariations()) { // Configure benchmark.