def test_given_hyperparameters_to_prophet_when_trainer_model_templates_called_then_hyperparameters_set_correctly(
temp_model_path, hyperparameters):
trainer = AutoTimeSeriesTrainer(path=temp_model_path, eval_metric="MAPE")
models = trainer.construct_model_templates(
hyperparameters=hyperparameters, )
for model in models:
for k, v in hyperparameters[model.name].items():
assert model._user_params[k] == v
def test_given_hyperparameters_and_custom_models_when_trainer_model_templates_called_then_hyperparameters_set_correctly(
temp_model_path, hyperparameters):
trainer = AutoTimeSeriesTrainer(path=temp_model_path, eval_metric="MAPE")
models = trainer.construct_model_templates(
hyperparameters=hyperparameters, )
for model in models:
if isinstance(model, GenericGluonTSModel):
model_hyperparam = next(
hyperparameters[m] for m in hyperparameters
if isinstance(m, GenericGluonTSModelFactory))
else:
model_hyperparam = hyperparameters[model.name]
for k, v in model_hyperparam.items():
assert model._user_params[k] == v