upgini 1.2.167a4220.dev4__tar.gz → 1.2.167a4220.dev5__tar.gz

{upgini-1.2.167a4220.dev4 → upgini-1.2.167a4220.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: upgini
-Version: 1.2.167a4220.dev4
+Version: 1.2.167a4220.dev5
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/

upgini-1.2.167a4220.dev5/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.167.a4220.dev5"

{upgini-1.2.167a4220.dev4 → upgini-1.2.167a4220.dev5}/src/upgini/features_enricher.py

@@ -1462,6 +1462,9 @@ class FeaturesEnricher(TransformerMixin):
         progress_bar: bool = True,
         progress_callback: Callable | None = None,
     ):
+        if self._has_single_ensemble_score():
+            self.logger.info("Skip OOT PSI stability check for single ensemble score")
+            return
         search_keys = self.search_keys.copy()
         search_keys.update(self._get_autodetected_search_keys())
         validated_X, _, validated_eval_set = self._validate_train_eval(X, y, eval_set)

{upgini-1.2.167a4220.dev4 → upgini-1.2.167a4220.dev5}/src/upgini/report/html.py

@@ -79,7 +79,7 @@ def _samples(data: ReportData) -> list[dict]:
         {
             "id": _slug(name),
             "label": name,
-            "caption": "",
+            "caption": "OOF validation" if index == 0 else "",
             "color": _SAMPLE_COLORS[index % len(_SAMPLE_COLORS)],
         }
         for index, name in enumerate(names)

{upgini-1.2.167a4220.dev4 → upgini-1.2.167a4220.dev5}/src/upgini/report/stats.py

@@ -344,10 +344,17 @@ def _score_psi(names: list[str], scored_samples: dict[str, pd.DataFrame], timeli
     if not timeline or not scored_samples:
         return []
     train = scored_samples.get(names[0]) if names else None
-    if train is None or SCORE not in train.columns:
+    if train is None or SCORE not in train.columns or DATE not in train.columns:
         return []
-    reference = pd.to_numeric(train[SCORE], errors="coerce").dropna()
-    if reference.empty:
+    train_scores = pd.to_numeric(train[SCORE], errors="coerce")
+    train_months = _month_keys(train[DATE])
+    reference = None
+    for month in timeline:
+        month_scores = train_scores[train_months == month].dropna()
+        if not month_scores.empty:
+            reference = month_scores
+            break
+    if reference is None:
         return []
     bins = _get_bin_edges(reference, HISTOGRAM_BINS)
     rows = []

{upgini-1.2.167a4220.dev4 → upgini-1.2.167a4220.dev5}/src/upgini/report/template.html

@@ -71,7 +71,7 @@
       </section>
 
       <section id="score-stability" class="section">
-        <div class="section-head"><div><h2>Score stability (PSI)</h2><p>Monthly PSI of scores · reference: full Train distribution · fixed reference bins.</p></div><div class="toolbar"><select id="psiSample" class="control" aria-label="Score stability sample"></select><button class="button ghost chart-export" data-chart="scorePsiChart">Save PNG</button></div></div>
+        <div class="section-head"><div><h2>Score stability (PSI)</h2><p>Monthly PSI of scores · reference: first period (baseline month) · fixed reference bins.</p></div><div class="toolbar"><select id="psiSample" class="control" aria-label="Score stability sample"></select><button class="button ghost chart-export" data-chart="scorePsiChart">Save PNG</button></div></div>
         <div class="panel chart-panel"><div class="threshold-legend" id="psiThresholdLegend"></div><div class="canvas-wrap"><canvas id="scorePsiChart" aria-label="Monthly score PSI" width="1596" height="560"></canvas><div class="chart-tooltip" id="psiTooltip"></div></div></div>
       </section>
 
@@ -628,7 +628,7 @@ const REPORT_DATA = __REPORT_DATA__;
     bindTooltip('scorePsiChart', 'psiTooltip', 'psi', i => {
       const cfg = D.scoreStability, s = cfg.bySample[document.getElementById('psiSample').value];
       return `<b>${MONTHS[i]} · ${selectedLabel('psiSample')}</b><br>PSI: ${fmt3(s.psi[i])}` +
-        `<br>Rows: ${num(s.rows[i])}<br>Score coverage: ${s.scoreCoverage[i].toFixed(1)}%<br>Reference: full Train distribution`;
+        `<br>Rows: ${num(s.rows[i])}<br>Score coverage: ${s.scoreCoverage[i].toFixed(1)}%<br>Reference: first period (baseline month)`;
     });
     bindTooltip('featureStabilityChart', 'featureTooltip', 'feature', i => {
       const cfg = D.featureStability, f = currentStabilityFeature(), s = cfg.bySample[f.name];

upgini-1.2.167a4220.dev4/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.167.a4220.dev4"