diff --git a/setup.py b/setup.py index 4ad992d..298c033 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import setup, find_packages setup( name='ttools', - version='0.7.92', + version='0.7.93', packages=find_packages(), install_requires=[ # list your dependencies here diff --git a/ttools/models.py b/ttools/models.py index c026ca7..2e0c28e 100644 --- a/ttools/models.py +++ b/ttools/models.py @@ -916,7 +916,7 @@ class LibraryTradingModel: y_fold_train, eval_set=[(X_fold_val, y_fold_val)], early_stopping_rounds=50, - verbose=False + verbose=10 ) # Calculate score @@ -1024,6 +1024,8 @@ class LibraryTradingModel: model = set_gpu_params(model) + model.set_params(verbosity=2) + #balance unbalanced classes, works for binary:logistics if self.config.n_classes == 2: n_0 = sum(y_train == 0) # 900