PyPI - wavetrainer - Versions diffs - 0.1.18__tar.gz → 0.1.19__tar.gz - Mend

@@ -5,6 +5,7 @@ import os
 from typing import Self
 import joblib  # type: ignore
+import numpy as np
 import optuna
 import pandas as pd
 from venn_abers import VennAbers  # type: ignore
@@ -55,18 +56,26 @@ class VennabersCalibrator(Calibrator):
             raise ValueError("vennabers is null")
         if y is None:
             raise ValueError("y is null")
-        prob_columns = [
-            x for x in df.columns.values if x.startswith(PROBABILITY_COLUMN_PREFIX)
-        ]
+        prob_columns = sorted(
+            [x for x in df.columns.values if x.startswith(PROBABILITY_COLUMN_PREFIX)]
+        )
+        probs = df[prob_columns].to_numpy()
         try:
-            vennabers.fit(df[prob_columns].to_numpy(), y.to_numpy())
+            vennabers.fit(probs, y.to_numpy())
         except IndexError:
             logging.error(df)
             raise
         return self
     def transform(self, df: pd.DataFrame) -> pd.DataFrame:
-        p_prime, _ = self._vennabers.predict_proba(df.to_numpy())
+        prob_columns = sorted(
+            [x for x in df.columns.values if x.startswith(PROBABILITY_COLUMN_PREFIX)]
+        )
+        probs = df[prob_columns].to_numpy()
+        p_prime, _ = self._vennabers.predict_proba(probs)
+        if np.mean(p_prime[:, 1] > 0.5) > 0.5 and np.mean(probs[:, 0] > 0.5) > 0.5:
+            print("⚠️ Warning: calibration seems inverted — flipping p_prime")
+            p_prime = p_prime[:, ::-1]
         for i in range(p_prime.shape[1]):
             prob = p_prime[:, i]
             df[f"{PROBABILITY_COLUMN_PREFIX}{i}"] = prob

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

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

@@ -40,7 +40,7 @@ class TestTrainer(unittest.TestCase):
     def test_trainer_dt_column(self):
         with tempfile.TemporaryDirectory() as tmpdir:
-            trainer = Trainer(tmpdir, walkforward_timedelta=datetime.timedelta(days=7), trials=5, dt_column="dt_column")
+            trainer = Trainer(tmpdir, walkforward_timedelta=datetime.timedelta(days=7), trials=5, dt_column="dt_column", allowed_models={"catboost"})
             x_data = [i for i in range(100)]
             x_index = [datetime.datetime(2022, 1, 1) + datetime.timedelta(days=i) for i in range(len(x_data))]
             df = pd.DataFrame(
@@ -55,6 +55,7 @@ class TestTrainer(unittest.TestCase):
                 },
                 index=df.index,
             )
+            y["y"] = y["y"].astype(bool)
             trainer.fit(df, y=y)
             df = trainer.transform(df)
             print("df:")

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

@@ -41,8 +41,8 @@ class CombinedReducer(Reducer):
             UnseenReducer(),
             NonNumericReducer(),
             PCAReducer(embedding_cols),
-            ConstantReducer(),
-            DuplicateReducer(),
+            # ConstantReducer(),
+            # DuplicateReducer(),
             CorrelationReducer(correlation_chunk_size=correlation_chunk_size),
             SmartCorrelationReducer(),
             # SelectBySingleFeaturePerformanceReducer(),

wavetrainer 0.1.18__tar.gz → 0.1.19__tar.gz

wavetrainer 0.1.18tar.gz → 0.1.19tar.gz