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

View File

@ -1,4 +1,5 @@
import logging
import sys
from pathlib import Path
from typing import Any, Dict
@ -48,6 +49,7 @@ class CatboostClassifier(BaseClassifierModel):
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

View File

@ -1,4 +1,5 @@
import logging
import sys
from pathlib import Path
from typing import Any, Dict
@ -47,6 +48,7 @@ class CatboostRegressor(BaseRegressionModel):
**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