wavetrainer 0.0.13__tar.gz → 0.0.15__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wavetrainer
-Version: 0.0.13
+Version: 0.0.15
 Summary: A library for automatically finding the optimal model within feature and hyperparameter space.
 Home-page: https://github.com/8W9aG/wavetrainer
 Author: Will Sackfield

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

@@ -23,7 +23,7 @@ def install_requires() -> typing.List[str]:
 
 setup(
     name='wavetrainer',
-    version='0.0.13',
+    version='0.0.15',
     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.13 → wavetrainer-0.0.15}/wavetrainer/__init__.py

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

{wavetrainer-0.0.13 → wavetrainer-0.0.15}/wavetrainer/model/catboost_model.py

@@ -78,14 +78,12 @@ class CatboostModel(Model):
     def set_options(self, trial: optuna.Trial | optuna.trial.FrozenTrial) -> None:
         self._iterations = trial.suggest_int(_ITERATIONS_KEY, 100, 10000)
         self._learning_rate = trial.suggest_float(_LEARNING_RATE_KEY, 0.001, 0.3)
-        self._depth = trial.suggest_int(_DEPTH_KEY, 1, 12)
+        self._depth = trial.suggest_int(_DEPTH_KEY, 1, 10)
         self._l2_leaf_reg = trial.suggest_float(_L2_LEAF_REG_KEY, 3.0, 50.0)
         self._boosting_type = trial.suggest_categorical(
             _BOOSTING_TYPE_KEY, ["Ordered", "Plain"]
         )
-        self._early_stopping_rounds = trial.suggest_int(
-            _EARLY_STOPPING_ROUNDS, 10, 1000
-        )
+        self._early_stopping_rounds = trial.suggest_int(_EARLY_STOPPING_ROUNDS, 10, 500)
 
     def load(self, folder: str) -> None:
         with open(

{wavetrainer-0.0.13 → wavetrainer-0.0.15}/wavetrainer/selector/selector.py

@@ -33,7 +33,7 @@ class Selector(Params, Fit):
 
     def set_options(self, trial: optuna.Trial | optuna.trial.FrozenTrial) -> None:
         self._feature_ratio = trial.suggest_float("feature_ratio", 0.0, 1.0)
-        self._steps = trial.suggest_int("steps", 1, 16)
+        self._steps = trial.suggest_int("steps", 1, 10)
 
     def load(self, folder: str) -> None:
         self._selector = joblib.load(os.path.join(folder, _SELECTOR_FILE))

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

@@ -317,7 +317,7 @@ class Trainer(Fit):
 
             last_processed_dt = None
             for count, test_idx in tqdm.tqdm(
-                enumerate(test_dt_index[test_dt_index >= start_test_index])
+                enumerate(dt_index[dt_index >= start_test_index])
             ):
                 if (
                     last_processed_dt is not None

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wavetrainer
-Version: 0.0.13
+Version: 0.0.15
 Summary: A library for automatically finding the optimal model within feature and hyperparameter space.
 Home-page: https://github.com/8W9aG/wavetrainer
 Author: Will Sackfield