adapt to the new strategy

This commit is contained in:
longyu
2023-01-31 19:44:32 +01:00
parent d595d9000d
commit f61bd549b2
4 changed files with 185 additions and 103 deletions

View File

@@ -49,6 +49,7 @@ class CatboostRegressor(BaseRegressionModel):
)
model.fit(X=train_data, eval_set=test_data, init_model=init_model,
verbose_eval=100,
log_cout=sys.stdout, log_cerr=sys.stderr)
return model

View File

@@ -62,6 +62,7 @@ class CatboostRegressorMultiTarget(BaseRegressionModel):
fit_params.append({
'eval_set': eval_sets[i], 'init_model': init_models[i],
'log_cout': sys.stdout, 'log_cerr': sys.stderr,
'verbose_eval': 100,
})
model = FreqaiMultiOutputRegressor(estimator=cbr)