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

Fix random test failure due to catboost bug 

https://github.com/catboost/catboost/issues/2195
This commit is contained in:
Matthias 2022-10-16 10:36:58 +02:00
parent dc50186d5b
commit 6cb14148aa
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
freqtrade/freqai/prediction_models/CatboostClassifier.py
View File

@ -1,4 +1,5 @@
import logging import logging
import sys
from pathlib import Path from pathlib import Path
from typing import Any, Dict from typing import Any, Dict
@ -48,6 +49,7 @@ class CatboostClassifier(BaseClassifierModel):
init_model = self.get_init_model(dk.pair) init_model = self.get_init_model(dk.pair)
cbr.fit(X=train_data, eval_set=test_data, init_model=init_model) cbr.fit(X=train_data, eval_set=test_data, init_model=init_model,
log_cout=sys.stdout, log_cerr=sys.stderr)
return cbr return cbr

4
freqtrade/freqai/prediction_models/CatboostRegressor.py
View File

@ -1,4 +1,5 @@
import logging import logging
import sys
from pathlib import Path from pathlib import Path
from typing import Any, Dict from typing import Any, Dict
@ -47,6 +48,7 @@ class CatboostRegressor(BaseRegressionModel):
**self.model_training_parameters, **self.model_training_parameters,
) )
model.fit(X=train_data, eval_set=test_data, init_model=init_model) model.fit(X=train_data, eval_set=test_data, init_model=init_model,
log_cout=sys.stdout, log_cerr=sys.stderr)
return model return model