From 874f90f211a7cb477fef9b498f96ea3401ddd3e6 Mon Sep 17 00:00:00 2001 From: capelastegui Date: Wed, 29 Jul 2020 15:33:55 +0100 Subject: [PATCH] feature #187; Metadata - incluye add_mult --- anticipy/forecast.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/anticipy/forecast.py b/anticipy/forecast.py index cf21390..f1b7ecc 100644 --- a/anticipy/forecast.py +++ b/anticipy/forecast.py @@ -1042,6 +1042,18 @@ def _get_use_calendar(df_y): return use_calendar +def _get_model_is_add_mult(model, add_mult_in, l_model_add): + """Return add_mult string for a model""" + # If add_mult_in is add or mult, return that + if add_mult_in in ('add', 'mult'): + return add_mult_in + # Otherwise, need to check if model is in list of additive models + if model in l_model_add: + return 'add' + else: + return 'mult' + + def run_forecast_single(df_y, l_model_trend=None, l_model_season=None, @@ -1293,12 +1305,16 @@ def run_forecast_single(df_y, l_model.sort() for model in l_model: - dict_fit_model = fit_model( model, df_y, freq, source, df_actuals=df_actuals_interpolated) df_metadata_tmp = dict_fit_model['metadata'] df_optimize_info = dict_fit_model['optimize_info'] + # Add add_mult to metadata + metadata_add_mult = _get_model_is_add_mult( + model, season_add_mult, l_model_add) + df_metadata_tmp['add_mult'] = metadata_add_mult + l_df_metadata += [df_metadata_tmp] l_df_optimize_info += [df_optimize_info] source_long = df_metadata_tmp.source_long.iloc[0]