From cc19b2f84ee72f6ecd9dfe2a3a51fe5a5b0fd2e7 Mon Sep 17 00:00:00 2001 From: Kevin Chen <13785749+Kevin-Chen0@users.noreply.github.com> Date: Sat, 3 Dec 2022 14:29:50 -0500 Subject: [PATCH] [fix] Renamed reamining self.config_season variables to self.config_seasonality in time_net.py (#1035) --- neuralprophet/time_net.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neuralprophet/time_net.py b/neuralprophet/time_net.py index 951820386..3fb4e1fb5 100644 --- a/neuralprophet/time_net.py +++ b/neuralprophet/time_net.py @@ -1258,8 +1258,8 @@ def _add_batch_regularizations(self, loss, epoch, progress): reg_loss += l_trend * reg_trend # Regularize seasonality: sparsify fourier term coefficients - if self.config_season: - l_season = self.config_season.reg_lambda + if self.config_seasonality: + l_season = self.config_seasonality.reg_lambda if self.season_dims is not None and l_season is not None and l_season > 0: for name in self.season_params.keys(): reg_season = utils.reg_func_season(self.season_params[name])