remove catboost GPU special case

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
pycaret · Mar 18, 2023 · ef12067 · ef12067
1 parent 09d2f40
commit ef12067
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 9 deletions.
diff --git a/pycaret/containers/models/classification.py b/pycaret/containers/models/classification.py
@@ -1424,9 +1424,7 @@ def __init__(self, experiment):
         # suppress output
         logging.getLogger("catboost").setLevel(logging.ERROR)
 
-        use_gpu = experiment.gpu_param == "force" or (
-            experiment.gpu_param and len(experiment.X_train) >= 50000
-        )
+        use_gpu = experiment.gpu_param
 
         args = {
             "random_state": experiment.seed,

diff --git a/pycaret/containers/models/regression.py b/pycaret/containers/models/regression.py
@@ -1802,9 +1802,7 @@ def __init__(self, experiment):
         # suppress output
         logging.getLogger("catboost").setLevel(logging.ERROR)
 
-        use_gpu = experiment.gpu_param == "force" or (
-            experiment.gpu_param and len(experiment.X_train) >= 50000
-        )
+        use_gpu = experiment.gpu_param
 
         args = {
             "random_state": experiment.seed,

diff --git a/pycaret/containers/models/time_series.py b/pycaret/containers/models/time_series.py
@@ -2448,9 +2448,7 @@ def __init__(self, experiment) -> None:
         # suppress output
         logging.getLogger("catboost").setLevel(logging.ERROR)
 
-        self.use_gpu = experiment.gpu_param == "force" or (
-            experiment.gpu_param and len(experiment.y_train) >= 50000
-        )
+        self.use_gpu = experiment.gpu_param
 
         super().__init__(experiment=experiment)