From c39c6d36a98306d65574db2141770d2842e68650 Mon Sep 17 00:00:00 2001 From: "Ahmed Hussein (amahussein)" Date: Tue, 11 Jun 2024 15:35:00 -0500 Subject: [PATCH] Enable xgboost prediction model by default Signed-off-by: Ahmed Hussein (amahussein) Fixes #1107 --- user_tools/src/spark_rapids_tools/cmdli/tools_cli.py | 4 ++-- user_tools/src/spark_rapids_tools/enums.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/user_tools/src/spark_rapids_tools/cmdli/tools_cli.py b/user_tools/src/spark_rapids_tools/cmdli/tools_cli.py index 7bd514b01..0ee8f2af3 100644 --- a/user_tools/src/spark_rapids_tools/cmdli/tools_cli.py +++ b/user_tools/src/spark_rapids_tools/cmdli/tools_cli.py @@ -95,8 +95,8 @@ def qualification(self, stage duration less than 25% of app duration and speedups greater than 1.3x. :param estimation_model: Model used to calculate the estimated GPU duration and cost savings. It accepts one of the following: - "xgboost": an XGBoost model for GPU duration estimation - "speedups": set by default. It uses a simple static estimated speedup per operator. + "xgboost": An XGBoost model for GPU duration estimation. Set by default + "speedups": It uses a simple static estimated speedup per operator. :param cpu_cluster_price: the CPU cluster hourly price provided by the user. :param estimated_gpu_cluster_price: the GPU cluster hourly price provided by the user. :param cpu_discount: A percent discount for the cpu cluster cost in the form of an integer value diff --git a/user_tools/src/spark_rapids_tools/enums.py b/user_tools/src/spark_rapids_tools/enums.py index 358f9570d..266ac50a0 100644 --- a/user_tools/src/spark_rapids_tools/enums.py +++ b/user_tools/src/spark_rapids_tools/enums.py @@ -167,4 +167,4 @@ class QualEstimationModel(EnumeratedType): @classmethod def get_default(cls): - return cls.SPEEDUPS + return cls.XGBOOST