dwker/ttools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
David Brazda
2024-11-27 09:20:24 +01:00
parent 0705b45351
commit 1b3f5b1b79
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -2,7 +2,7 @@ from setuptools import setup, find_packages
setup( setup(
name='ttools', name='ttools',
version='0.7.92', version='0.7.93',
packages=find_packages(), packages=find_packages(),
install_requires=[ install_requires=[
# list your dependencies here # list your dependencies here

4
ttools/models.py
View File

@ -916,7 +916,7 @@ class LibraryTradingModel:
y_fold_train, y_fold_train,
eval_set=[(X_fold_val, y_fold_val)], eval_set=[(X_fold_val, y_fold_val)],
early_stopping_rounds=50, early_stopping_rounds=50,
verbose=False verbose=10
) )
# Calculate score # Calculate score
@ -1024,6 +1024,8 @@ class LibraryTradingModel:
model = set_gpu_params(model) model = set_gpu_params(model)
model.set_params(verbosity=2)
#balance unbalanced classes, works for binary:logistics #balance unbalanced classes, works for binary:logistics
if self.config.n_classes == 2: if self.config.n_classes == 2:
n_0 = sum(y_train == 0) # 900 n_0 = sum(y_train == 0) # 900