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

fix

Browse Source
This commit is contained in:
David Brazda
2024-11-27 10:15:22 +01:00
parent 1b3f5b1b79
commit 623412406e
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.93', version='0.7.94',
packages=find_packages(), packages=find_packages(),
install_requires=[ install_requires=[
# list your dependencies here # list your dependencies here

2
ttools/models.py
View File

@ -903,6 +903,8 @@ class LibraryTradingModel:
else: else:
model = XGBRegressor(**params) model = XGBRegressor(**params)
model = set_gpu_params(model)
# Handle class imbalance for binary classification # Handle class imbalance for binary classification
if self.config.n_classes == 2: if self.config.n_classes == 2:
n_0 = sum(y_fold_train == 0) n_0 = sum(y_fold_train == 0)