update
This commit is contained in:
2
setup.py
2
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
|
||||
|
||||
@ -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
|
||||
|
||||
Reference in New Issue
Block a user