adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove verbosity params

Browse Source
This commit is contained in:
Emre 2022-09-09 00:11:43 +03:00 committed by Robert Caulk
parent df6e43d2c5
commit acb410a0de
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
freqtrade/freqai/prediction_models/XGBoostRegressor.py
View File

@ -1,7 +1,7 @@
import logging
from typing import Any, Dict
import xgboost as xgb
from xgboost import XGBRegressor
from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
from freqtrade.freqai.prediction_models.BaseRegressionModel import BaseRegressionModel
@ -24,9 +24,6 @@ class XGBoostRegressor(BaseRegressionModel):
all the training and test data/labels.
"""
xgb.set_config(verbosity=2)
xgb.config_context(verbosity=2)
X = data_dictionary["train_features"]
y = data_dictionary["train_labels"]
@ -39,7 +36,7 @@ class XGBoostRegressor(BaseRegressionModel):
xgb_model = self.get_init_model(dk.pair)
model = xgb.XGBRegressor(**self.model_training_parameters)
model = XGBRegressor(**self.model_training_parameters)
model.fit(X=X, y=y, sample_weight=sample_weight, eval_set=eval_set, xgb_model=xgb_model)