adapt to the new strategy
This commit is contained in:
@@ -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
|
||||
|
@@ -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)
|
||||
|
Reference in New Issue
Block a user