upgini 1.1.197__tar.gz → 1.1.199__tar.gz

{upgini-1.1.197/src/upgini.egg-info → upgini-1.1.199}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: upgini
-Version: 1.1.197
+Version: 1.1.199
 Summary: Intelligent data search & enrichment for Machine Learning
 Home-page: https://upgini.com/
 Author: Upgini Developers
@@ -96,7 +96,12 @@ License-File: LICENSE
     <td><a href="https://en.wikipedia.org/wiki/Regression_analysis">☑️ regression</a></td>
     <td><a href="https://en.wikipedia.org/wiki/Time_series#Prediction_and_forecasting">☑️ time series prediction</a></td>
   </tr>
-</table>
+</table>   
+
+⭐️ [Simple Drag & Drop Search UI](https://appwidget-uragwvgykrk4sbmropphpy.streamlit.app/~/+/):   
+<a href="https://appwidget-uragwvgykrk4sbmropphpy.streamlit.app/~/+/">
+<img width="710" alt="Drag & Drop Search UI" src="https://github.com/upgini/upgini/assets/95645411/36b6460c-51f3-400e-9f04-445b938bf45e">
+</a>
 
 
 ## 🌎 Connected data sources and coverage

{upgini-1.1.197 → upgini-1.1.199}/README.md

@@ -66,7 +66,12 @@
     <td><a href="https://en.wikipedia.org/wiki/Regression_analysis">☑️ regression</a></td>
     <td><a href="https://en.wikipedia.org/wiki/Time_series#Prediction_and_forecasting">☑️ time series prediction</a></td>
   </tr>
-</table>
+</table>   
+
+⭐️ [Simple Drag & Drop Search UI](https://appwidget-uragwvgykrk4sbmropphpy.streamlit.app/~/+/):   
+<a href="https://appwidget-uragwvgykrk4sbmropphpy.streamlit.app/~/+/">
+<img width="710" alt="Drag & Drop Search UI" src="https://github.com/upgini/upgini/assets/95645411/36b6460c-51f3-400e-9f04-445b938bf45e">
+</a>
 
 
 ## 🌎 Connected data sources and coverage

{upgini-1.1.197 → upgini-1.1.199}/setup.py

@@ -40,7 +40,7 @@ def send_log(msg: str):
 
 
 here = Path(__file__).parent.resolve()
-version = "1.1.197"
+version = "1.1.199"
 try:
     send_log(f"Start setup PyLib version {version}")
     setup(

{upgini-1.1.197 → upgini-1.1.199}/src/upgini/features_enricher.py

@@ -13,7 +13,7 @@ from typing import Any, Callable, Dict, List, Optional, Set, Tuple, Union
 
 import numpy as np
 import pandas as pd
-from pandas.api.types import is_string_dtype
+from pandas.api.types import is_numeric_dtype, is_string_dtype
 from scipy.stats import ks_2samp
 from sklearn.base import TransformerMixin
 from sklearn.exceptions import NotFittedError
@@ -946,7 +946,9 @@ class FeaturesEnricher(TransformerMixin):
                         bundle.get("quality_metrics_rows_header"): _num_samples(fitting_X),
                         # bundle.get("quality_metrics_match_rate_header"): self._search_task.initial_max_hit_rate_v2(),
                     }
-                    if model_task_type in [ModelTaskType.BINARY, ModelTaskType.REGRESSION]:
+                    if model_task_type in [ModelTaskType.BINARY, ModelTaskType.REGRESSION] and is_numeric_dtype(
+                        y_sorted
+                    ):
                         train_metrics[bundle.get("quality_metrics_mean_target_header")] = round(y_sorted.mean(), 4)
                     if etalon_metric is not None:
                         train_metrics[bundle.get("quality_metrics_baseline_header").format(metric)] = etalon_metric
@@ -1008,7 +1010,9 @@ class FeaturesEnricher(TransformerMixin):
                                 bundle.get("quality_metrics_rows_header"): _num_samples(eval_X_sorted),
                                 # bundle.get("quality_metrics_match_rate_header"): eval_hit_rate,
                             }
-                            if model_task_type in [ModelTaskType.BINARY, ModelTaskType.REGRESSION]:
+                            if model_task_type in [ModelTaskType.BINARY, ModelTaskType.REGRESSION] and is_numeric_dtype(
+                                eval_y_sorted
+                            ):
                                 eval_metrics[bundle.get("quality_metrics_mean_target_header")] = round(
                                     eval_y_sorted.mean(), 4
                                 )
@@ -1204,7 +1208,7 @@ class FeaturesEnricher(TransformerMixin):
             self.logger.info("No external features selected. So use only input datasets for metrics calculation")
             X_sampled, search_keys = self._extend_x(validated_X, is_demo_dataset)
             y_sampled = validated_y
-            enriched_X = validated_X
+            enriched_X = X_sampled
             if eval_set is not None:
                 for idx in range(len(eval_set)):
                     eval_X_sampled, _ = self._extend_x(eval_set[idx][0], is_demo_dataset)
@@ -2315,7 +2319,7 @@ class FeaturesEnricher(TransformerMixin):
             do_without_pandas_limits(print_datasets_sample)
 
             maybe_date_col = self._get_date_column(self.search_keys)
-            if X is not None and maybe_date_col is not None:
+            if X is not None and maybe_date_col is not None and maybe_date_col in X.columns:
                 min_date = X[maybe_date_col].min()
                 max_date = X[maybe_date_col].max()
                 self.logger.info(f"Dates interval is ({min_date}, {max_date})")

{upgini-1.1.197 → upgini-1.1.199/src/upgini.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: upgini
-Version: 1.1.197
+Version: 1.1.199
 Summary: Intelligent data search & enrichment for Machine Learning
 Home-page: https://upgini.com/
 Author: Upgini Developers
@@ -96,7 +96,12 @@ License-File: LICENSE
     <td><a href="https://en.wikipedia.org/wiki/Regression_analysis">☑️ regression</a></td>
     <td><a href="https://en.wikipedia.org/wiki/Time_series#Prediction_and_forecasting">☑️ time series prediction</a></td>
   </tr>
-</table>
+</table>   
+
+⭐️ [Simple Drag & Drop Search UI](https://appwidget-uragwvgykrk4sbmropphpy.streamlit.app/~/+/):   
+<a href="https://appwidget-uragwvgykrk4sbmropphpy.streamlit.app/~/+/">
+<img width="710" alt="Drag & Drop Search UI" src="https://github.com/upgini/upgini/assets/95645411/36b6460c-51f3-400e-9f04-445b938bf45e">
+</a>
 
 
 ## 🌎 Connected data sources and coverage