wavetrainer 0.0.11__tar.gz → 0.0.13__tar.gz

{wavetrainer-0.0.11/wavetrainer.egg-info → wavetrainer-0.0.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wavetrainer
-Version: 0.0.11
+Version: 0.0.13
 Summary: A library for automatically finding the optimal model within feature and hyperparameter space.
 Home-page: https://github.com/8W9aG/wavetrainer
 Author: Will Sackfield
@@ -21,6 +21,7 @@ Requires-Dist: scipy>=1.15.2
 Requires-Dist: catboost>=1.2.7
 Requires-Dist: venn-abers>=1.4.6
 Requires-Dist: mapie>=0.9.2
+Requires-Dist: pytz>=2025.1
 
 # wavetrainer
 
@@ -48,6 +49,7 @@ Python 3.11.6:
 - [catboost](https://catboost.ai/)
 - [venn-abers](https://github.com/ip200/venn-abers)
 - [mapie](https://mapie.readthedocs.io/en/stable/)
+- [pytz](https://pythonhosted.org/pytz/)
 
 ## Raison D'être :thought_balloon:
 

{wavetrainer-0.0.11 → wavetrainer-0.0.13}/README.md

@@ -24,6 +24,7 @@ Python 3.11.6:
 - [catboost](https://catboost.ai/)
 - [venn-abers](https://github.com/ip200/venn-abers)
 - [mapie](https://mapie.readthedocs.io/en/stable/)
+- [pytz](https://pythonhosted.org/pytz/)
 
 ## Raison D'être :thought_balloon:
 

wavetrainer-0.0.11/wavetrainer.egg-info/requires.txt → wavetrainer-0.0.13/requirements.txt

@@ -8,3 +8,4 @@ scipy>=1.15.2
 catboost>=1.2.7
 venn-abers>=1.4.6
 mapie>=0.9.2
+pytz>=2025.1

{wavetrainer-0.0.11 → wavetrainer-0.0.13}/setup.py

@@ -23,7 +23,7 @@ def install_requires() -> typing.List[str]:
 
 setup(
     name='wavetrainer',
-    version='0.0.11',
+    version='0.0.13',
     description='A library for automatically finding the optimal model within feature and hyperparameter space.',
     long_description=long_description,
     long_description_content_type='text/markdown',

{wavetrainer-0.0.11 → wavetrainer-0.0.13}/wavetrainer/__init__.py

@@ -2,5 +2,5 @@
 
 from .create import create
 
-__VERSION__ = "0.0.11"
+__VERSION__ = "0.0.13"
 __all__ = ("create",)

{wavetrainer-0.0.11 → wavetrainer-0.0.13}/wavetrainer/create.py

@@ -14,6 +14,7 @@ def create(
     validation_size: float | datetime.timedelta | None = None,
     dt_column: str | None = None,
     max_train_timeout: datetime.timedelta | None = None,
+    cutoff_dt: datetime.datetime | None = None,
 ) -> Trainer:
     """Create a trainer."""
     return Trainer(
@@ -23,4 +24,5 @@ def create(
         validation_size=validation_size,
         dt_column=dt_column,
         max_train_timeout=max_train_timeout,
+        cutoff_dt=cutoff_dt,
     )

{wavetrainer-0.0.11 → wavetrainer-0.0.13}/wavetrainer/trainer.py

@@ -41,7 +41,7 @@ _DT_COLUMN_KEY = "dt_column"
 class Trainer(Fit):
     """A class for training and predicting from an array of data."""
 
-    # pylint: disable=too-many-arguments,too-many-positional-arguments,too-many-statements,too-many-locals,too-many-branches
+    # pylint: disable=too-many-arguments,too-many-positional-arguments,too-many-statements,too-many-locals,too-many-branches,too-many-instance-attributes
 
     def __init__(
         self,
@@ -52,6 +52,7 @@ class Trainer(Fit):
         validation_size: float | datetime.timedelta | None = None,
         dt_column: str | None = None,
         max_train_timeout: datetime.timedelta | None = None,
+        cutoff_dt: datetime.datetime | None = None,
     ):
         tqdm.tqdm.pandas()
 
@@ -65,6 +66,8 @@ class Trainer(Fit):
             test_size = 0.15
         if validation_size is None:
             validation_size = 0.15
+        if cutoff_dt is None:
+            cutoff_dt = datetime.datetime.now()
 
         params_file = os.path.join(self._folder, _PARAMS_FILENAME)
         if walkforward_timedelta is None:
@@ -139,6 +142,7 @@ class Trainer(Fit):
         self._validation_size = validation_size
         self._dt_column = dt_column
         self._max_train_timeout = max_train_timeout
+        self._cutoff_dt = cutoff_dt
 
     def _provide_study(self, column: str) -> optuna.Study:
         storage_name = f"sqlite:///{self._folder}/{column}/{_STUDYDB_FILENAME}"
@@ -172,6 +176,9 @@ class Trainer(Fit):
             if self._dt_column is None
             else pd.DatetimeIndex(pd.to_datetime(df[self._dt_column]))
         )
+        df = df[dt_index < self._cutoff_dt]  # type: ignore
+        y = y.iloc[: len(df)]
+        dt_index = dt_index[: len(df)]
 
         def _fit_column(y_series: pd.Series):
             column_dir = os.path.join(self._folder, str(y_series.name))
@@ -187,6 +194,7 @@ class Trainer(Fit):
                 save: bool,
                 split_idx: datetime.datetime,
             ) -> float:
+                print(f"Beginning trial for: {split_idx.isoformat()}")
                 trial.set_user_attr(_IDX_USR_ATTR_KEY, split_idx.isoformat())
 
                 train_dt_index = dt_index[: len(x)]
@@ -326,6 +334,7 @@ class Trainer(Fit):
                         found = True
                         break
                 if found:
+                    last_processed_dt = test_dt
                     continue
 
                 test_df = df.iloc[: train_len + count + test_len]
@@ -349,6 +358,7 @@ class Trainer(Fit):
                     )
 
                 _fit(study.best_trial, test_df, test_series, True, test_idx)
+                last_processed_dt = test_idx
 
         if isinstance(y, pd.Series):
             _fit_column(y)

{wavetrainer-0.0.11 → wavetrainer-0.0.13/wavetrainer.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wavetrainer
-Version: 0.0.11
+Version: 0.0.13
 Summary: A library for automatically finding the optimal model within feature and hyperparameter space.
 Home-page: https://github.com/8W9aG/wavetrainer
 Author: Will Sackfield
@@ -21,6 +21,7 @@ Requires-Dist: scipy>=1.15.2
 Requires-Dist: catboost>=1.2.7
 Requires-Dist: venn-abers>=1.4.6
 Requires-Dist: mapie>=0.9.2
+Requires-Dist: pytz>=2025.1
 
 # wavetrainer
 
@@ -48,6 +49,7 @@ Python 3.11.6:
 - [catboost](https://catboost.ai/)
 - [venn-abers](https://github.com/ip200/venn-abers)
 - [mapie](https://mapie.readthedocs.io/en/stable/)
+- [pytz](https://pythonhosted.org/pytz/)
 
 ## Raison D'être :thought_balloon:
 

wavetrainer-0.0.11/requirements.txt → wavetrainer-0.0.13/wavetrainer.egg-info/requires.txt

@@ -7,4 +7,5 @@ numpy>=1.26.4
 scipy>=1.15.2
 catboost>=1.2.7
 venn-abers>=1.4.6
-mapie>=0.9.2
+mapie>=0.9.2
+pytz>=2025.1