From 1b3f5b1b7998ab393a44bd3c03430ff67b0c365a Mon Sep 17 00:00:00 2001 From: David Brazda Date: Wed, 27 Nov 2024 09:20:24 +0100 Subject: [PATCH] update --- setup.py | 2 +- ttools/models.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) 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