autogluon.timeseries 0.7.0b20230302__py3-none-any.whl → 0.7.0b20230303__py3-none-any.whl

autogluon/timeseries/dataset/ts_dataframe.py

@@ -8,6 +8,7 @@ from typing import Any, Optional, Tuple, Type
 
 import numpy as np
 import pandas as pd
+from joblib.parallel import Parallel, delayed
 from pandas.core.internals import ArrayManager, BlockManager
 
 from autogluon.common.loaders import load_pd
@@ -258,24 +259,27 @@ class TimeSeriesDataFrame(pd.DataFrame):
             raise ValueError(f"all entries in index `{ITEMID}` must be of integer or string dtype")
 
     @classmethod
-    def _construct_pandas_frame_from_iterable_dataset(cls, iterable_dataset: Iterable) -> pd.DataFrame:
-        cls._validate_iterable(iterable_dataset)
-
-        all_ts = []
-        for i, ts in enumerate(iterable_dataset):
+    def _construct_pandas_frame_from_iterable_dataset(
+        cls, iterable_dataset: Iterable, num_cpus: int = -1
+    ) -> pd.DataFrame:
+        def load_single_item(item_id: int, ts: dict) -> pd.DataFrame:
             start_timestamp = ts["start"]
             freq = start_timestamp.freq
             if isinstance(start_timestamp, pd.Period):
                 start_timestamp = start_timestamp.to_timestamp(how="S")
             target = ts["target"]
             datetime_index = tuple(pd.date_range(start_timestamp, periods=len(target), freq=freq))
-            idx = pd.MultiIndex.from_product([(i,), datetime_index], names=[ITEMID, TIMESTAMP])
-            ts_df = pd.Series(target, name="target", index=idx).to_frame()
-            all_ts.append(ts_df)
+            idx = pd.MultiIndex.from_product([(item_id,), datetime_index], names=[ITEMID, TIMESTAMP])
+            return pd.Series(target, name="target", index=idx).to_frame()
+
+        cls._validate_iterable(iterable_dataset)
+        all_ts = Parallel(n_jobs=num_cpus)(
+            delayed(load_single_item)(item_id, ts) for item_id, ts in enumerate(iterable_dataset)
+        )
         return pd.concat(all_ts)
 
     @classmethod
-    def from_iterable_dataset(cls, iterable_dataset: Iterable) -> pd.DataFrame:
+    def from_iterable_dataset(cls, iterable_dataset: Iterable, num_cpus: int = -1) -> pd.DataFrame:
         """Construct a ``TimeSeriesDataFrame`` from an Iterable of dictionaries each of which
         represent a single time series.
 
@@ -294,13 +298,15 @@ class TimeSeriesDataFrame(pd.DataFrame):
                     {"target": [3, 4, 5], "start": pd.Timestamp("01-01-2019", freq='D')},
                     {"target": [6, 7, 8], "start": pd.Timestamp("01-01-2019", freq='D')}
                 ]
+        num_cpus : int, default = -1
+            Number of CPU cores used to process the iterable dataset in parallel. Set to -1 to use all cores.
 
         Returns
         -------
         ts_df: TimeSeriesDataFrame
             A data frame in TimeSeriesDataFrame format.
         """
-        return cls(cls._construct_pandas_frame_from_iterable_dataset(iterable_dataset))
+        return cls(cls._construct_pandas_frame_from_iterable_dataset(iterable_dataset, num_cpus=num_cpus))
 
     @classmethod
     def _load_data_frame_from_file(cls, path: str) -> pd.DataFrame:

autogluon/timeseries/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '0.7.0b20230302'
+__version__ = '0.7.0b20230303'
 __lite__ = False

{autogluon.timeseries-0.7.0b20230302.dist-info → autogluon.timeseries-0.7.0b20230303.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.timeseries
-Version: 0.7.0b20230302
+Version: 0.7.0b20230303
 Summary: AutoML for Image, Text, and Tabular Data
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -45,12 +45,12 @@ Requires-Dist: networkx (<3.0,>=2.3)
 Requires-Dist: statsforecast (<1.5,>=1.4.0)
 Requires-Dist: tqdm (<5,>=4.38)
 Requires-Dist: ujson (<6,>=5)
-Requires-Dist: autogluon.core[raytune] (==0.7.0b20230302)
-Requires-Dist: autogluon.common (==0.7.0b20230302)
-Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost] (==0.7.0b20230302)
+Requires-Dist: autogluon.core[raytune] (==0.7.0b20230303)
+Requires-Dist: autogluon.common (==0.7.0b20230303)
+Requires-Dist: autogluon.tabular[catboost,lightgbm,xgboost] (==0.7.0b20230303)
 Provides-Extra: all
-Requires-Dist: tbats (<2,>=1.1) ; extra == 'all'
 Requires-Dist: sktime (<0.16,>=0.14) ; extra == 'all'
+Requires-Dist: tbats (<2,>=1.1) ; extra == 'all'
 Requires-Dist: pmdarima (<1.9,>=1.8.2) ; extra == 'all'
 Provides-Extra: sktime
 Requires-Dist: sktime (<0.16,>=0.14) ; extra == 'sktime'

{autogluon.timeseries-0.7.0b20230302.dist-info → autogluon.timeseries-0.7.0b20230303.dist-info}/RECORD

@@ -1,14 +1,14 @@
-autogluon.timeseries-0.7.0b20230302-py3.8-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.timeseries-0.7.0b20230303-py3.8-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/timeseries/__init__.py,sha256=Gj3vH2ndDbIz30cC_MoP60TEdrkPUjoHqzTGG9k1lhk,1298
 autogluon/timeseries/evaluator.py,sha256=Xsr01q1viDE-n_UUiYcVblyZX0I8hvMkOgNu4mu-f_c,10641
 autogluon/timeseries/learner.py,sha256=P_msIgQaskfuUwuPaUOGwr9VE-IJrJHetdM9irs6U7k,10468
 autogluon/timeseries/predictor.py,sha256=Jg2Zhg1nLjYGqG741TXTKYk7IVNY4I2QgzRC25Y7g8M,40636
 autogluon/timeseries/splitter.py,sha256=s5S3CeJxcUfZrl7PSXjzubE06bgB8J8uUT8EywSwtYQ,9252
-autogluon/timeseries/version.py,sha256=KZ2gLHJghvdHAOjDzignbwQX3lXY_65WRLeGyUdaAW4,90
+autogluon/timeseries/version.py,sha256=tGDT681gW3NtzFXd-CGS1stLwVKrIlCt0Efc83z3iDQ,90
 autogluon/timeseries/configs/__init__.py,sha256=BTtHIPCYeGjqgOcvqb8qPD4VNX-ICKOg6wnkew1cPOE,98
 autogluon/timeseries/configs/presets_configs.py,sha256=chnf3EvCeMP1zUg6aYkz499zrOiMlLul4ETJnNVOslA,799
 autogluon/timeseries/dataset/__init__.py,sha256=UvnhAN5tjgxXTHoZMQDy64YMDj4Xxa68yY7NP4vAw0o,81
-autogluon/timeseries/dataset/ts_dataframe.py,sha256=rVIyoyGT86R3lehjV3rZmejyz8VGMfBt9c6judeYBw8,33264
+autogluon/timeseries/dataset/ts_dataframe.py,sha256=07dmHBpttBF0Mih-AgKXWbQDguVlr4Qpl2AIxBrgUfM,33654
 autogluon/timeseries/models/__init__.py,sha256=LFMK5eYZD2t2lvSFX352wUTHxdDi6YLDMAltnjsG2VQ,624
 autogluon/timeseries/models/presets.py,sha256=Gb-FHhJVGPPLHd74bmsmQo2wbFVwtU1oWcTV6IC0eQ0,9475
 autogluon/timeseries/models/abstract/__init__.py,sha256=DyBTMHj4b4D35aiR83FcCR0_MvPHdqgpmEbdiPoVwcw,168
@@ -44,11 +44,11 @@ autogluon/timeseries/utils/hashing.py,sha256=tcjTgJ0vA4WV2-WnvF8iW7la5duwCpSUKk3
 autogluon/timeseries/utils/random.py,sha256=W300Co4eG4txIsnj8pdFFia7DhyVJ5oXWKOEjnGDuVA,344
 autogluon/timeseries/utils/seasonality.py,sha256=p9mtahWOtDhHUjeGECUJA0VAKeLkZGZbj070dEqMTJQ,652
 autogluon/timeseries/utils/warning_filters.py,sha256=09a3JgFoVG-bfY92bvr80WfSdLI7CxG1jI546WKPUnw,2330
-autogluon.timeseries-0.7.0b20230302.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.timeseries-0.7.0b20230302.dist-info/METADATA,sha256=v8Utomx9yAvIupsWmNrG8tmM9weMYM5K8yx6Wim4omg,12699
-autogluon.timeseries-0.7.0b20230302.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.timeseries-0.7.0b20230302.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-autogluon.timeseries-0.7.0b20230302.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-0.7.0b20230302.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.timeseries-0.7.0b20230302.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.timeseries-0.7.0b20230302.dist-info/RECORD,,
+autogluon.timeseries-0.7.0b20230303.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.timeseries-0.7.0b20230303.dist-info/METADATA,sha256=r5GuMPbZ0qCMgrYsy4ef8gih3wYWJT5ZHQoy6neGOEk,12699
+autogluon.timeseries-0.7.0b20230303.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.timeseries-0.7.0b20230303.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+autogluon.timeseries-0.7.0b20230303.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-0.7.0b20230303.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.timeseries-0.7.0b20230303.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.timeseries-0.7.0b20230303.dist-info/RECORD,,