[fix] Renamed reamining self.config_season variables to self.config_s…

…easonality in time_net.py (#1035)
ourownstory · Dec 3, 2022 · cc19b2f · cc19b2f
1 parent 2154c26
commit cc19b2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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])