autogluon.timeseries 1.1.2b20240819__tar.gz → 1.1.2b20240820__tar.gz

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240820}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.1.2b20240819
+Version: 1.1.2b20240820
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240820}/src/autogluon/timeseries/models/local/abstract_local_model.py

@@ -222,18 +222,22 @@ def seasonal_naive_forecast(
 ) -> pd.DataFrame:
     """Generate seasonal naive forecast, predicting the last observed value from the same period."""
 
-    def numpy_ffill(arr: np.ndarray) -> np.ndarray:
-        """Fast implementation of forward fill in numpy."""
+    def numpy_fillna(arr: np.ndarray) -> np.ndarray:
+        """Fast implementation of forward fill + avg fill in numpy."""
+        # First apply forward fill
         idx = np.arange(len(arr))
         mask = np.isnan(arr)
         idx[mask] = 0
-        return arr[np.maximum.accumulate(idx)]
+        arr_filled = arr[np.maximum.accumulate(idx)]
+        # Leading NaNs are filled with the mean
+        arr_filled[np.isnan(arr_filled)] = np.nanmean(arr_filled)
+        return arr_filled
 
     forecast = {}
     # At least seasonal_period + 2 values are required to compute sigma for seasonal naive
     if len(target) > seasonal_period + 1 and seasonal_period > 1:
         if np.isnan(target[-(seasonal_period + 2) :]).any():
-            target = numpy_ffill(target)
+            target = numpy_fillna(target)
 
         indices = [len(target) - seasonal_period + k % seasonal_period for k in range(prediction_length)]
         forecast["mean"] = target[indices]

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240820}/src/autogluon/timeseries/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '1.1.2b20240819'
+__version__ = '1.1.2b20240820'
 __lite__ = False

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240820}/src/autogluon.timeseries.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 1.1.2b20240819
+Version: 1.1.2b20240820
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.timeseries-1.1.2b20240819 → autogluon.timeseries-1.1.2b20240820}/src/autogluon.timeseries.egg-info/requires.txt

@@ -16,9 +16,9 @@ fugue>=0.9.0
 tqdm<5,>=4.38
 orjson~=3.9
 tensorboard<3,>=2.9
-autogluon.core[raytune]==1.1.2b20240819
-autogluon.common==1.1.2b20240819
-autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20240819
+autogluon.core[raytune]==1.1.2b20240820
+autogluon.common==1.1.2b20240820
+autogluon.tabular[catboost,lightgbm,xgboost]==1.1.2b20240820
 
 [all]
 optimum[onnxruntime]<1.19,>=1.17