Fix sys.stdout bug for CatboostRegressorMultiTarget

This commit is contained in:
Matthias 2022-10-17 19:48:27 +02:00
parent 6cb14148aa
commit 441032be25
1 changed files with 5 additions and 2 deletions

View File

@ -1,4 +1,5 @@
import logging
import sys
from pathlib import Path
from typing import Any, Dict
@ -58,8 +59,10 @@ class CatboostRegressorMultiTarget(BaseRegressionModel):
fit_params = []
for i in range(len(eval_sets)):
fit_params.append(
{'eval_set': eval_sets[i], 'init_model': init_models[i]})
fit_params.append({
'eval_set': eval_sets[i], 'init_model': init_models[i],
'log_cout': sys.stdout, 'log_cerr': sys.stderr,
})
model = FreqaiMultiOutputRegressor(estimator=cbr)
thread_training = self.freqai_info.get('multitarget_parallel_training', False)