fix

2024-11-27 10:15:22 +01:00
parent 1b3f5b1b79
commit 623412406e
2 changed files with 4 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@ -2,7 +2,7 @@ from setuptools import setup, find_packages

 setup(
    name='ttools',
-    version='0.7.93',
+    version='0.7.94',
    packages=find_packages(),
    install_requires=[
        # list your dependencies here
--- a/ttools/models.py
+++ b/ttools/models.py
@ -902,7 +902,9 @@ class LibraryTradingModel:
                        model.set_params(num_class=self.config.n_classes)
                else:
                    model = XGBRegressor(**params)
-                
+
+                model = set_gpu_params(model)
+
                # Handle class imbalance for binary classification
                if self.config.n_classes == 2:
                    n_0 = sum(y_fold_train == 0)