upgini 1.1.166a2__tar.gz → 1.1.168__tar.gz

{upgini-1.1.166a2/src/upgini.egg-info → upgini-1.1.168}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: upgini
-Version: 1.1.166a2
+Version: 1.1.168
 Summary: Intelligent data search & enrichment for Machine Learning
 Home-page: https://upgini.com/
 Author: Upgini Developers

{upgini-1.1.166a2 → upgini-1.1.168}/setup.py

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

{upgini-1.1.166a2 → upgini-1.1.168}/src/upgini/dataset.py

@@ -45,11 +45,11 @@ from upgini.utils.warning_counter import WarningCounter
 
 class Dataset:  # (pd.DataFrame):
     MIN_ROWS_COUNT = 100
-    MAX_ROWS = 300_000
+    MAX_ROWS = 200_000
     FIT_SAMPLE_ROWS = 200_000
     FIT_SAMPLE_THRESHOLD = 200_000
-    FIT_SAMPLE_WITH_EVAL_SET_ROWS = 300_000
-    FIT_SAMPLE_WITH_EVAL_SET_THRESHOLD = 300_000
+    FIT_SAMPLE_WITH_EVAL_SET_ROWS = 200_000
+    FIT_SAMPLE_WITH_EVAL_SET_THRESHOLD = 200_000
     MIN_SAMPLE_THRESHOLD = 20_000
     IMBALANCE_THESHOLD = 0.4
     MIN_TARGET_CLASS_ROWS = 100

{upgini-1.1.166a2 → upgini-1.1.168}/src/upgini/features_enricher.py

@@ -111,6 +111,7 @@ class FeaturesEnricher(TransformerMixin):
     RANDOM_STATE = 42
     CALCULATE_METRICS_THRESHOLD = 50_000_000
     CALCULATE_METRICS_MIN_THRESHOLD = 500
+    GENERATE_FEATURES_LIMIT = 10
     EMPTY_FEATURES_INFO = pd.DataFrame(
         columns=[
             bundle.get("features_info_provider"),
@@ -211,8 +212,8 @@ class FeaturesEnricher(TransformerMixin):
         self.generate_features = generate_features
         self.round_embeddings = round_embeddings
         if generate_features is not None:
-            if len(generate_features) > 2:
-                msg = bundle.get("too_many_generate_features")
+            if len(generate_features) > self.GENERATE_FEATURES_LIMIT:
+                msg = bundle.get("too_many_generate_features").format(self.GENERATE_FEATURES_LIMIT)
                 self.logger.error(msg)
                 raise ValidationError(msg)
             self.runtime_parameters.properties["generate_features"] = ",".join(generate_features)
@@ -1755,7 +1756,7 @@ class FeaturesEnricher(TransformerMixin):
             and len(self._search_task.unused_features_for_generation) > 0
         ):
             unused_features_for_generation = [
-                dataset.columns_renaming.get(col) for col in self._search_task.unused_features_for_generation
+                dataset.columns_renaming.get(col) or col for col in self._search_task.unused_features_for_generation
             ]
             msg = bundle.get("features_not_generated").format(unused_features_for_generation)
             self.logger.warning(msg)

{upgini-1.1.166a2 → upgini-1.1.168}/src/upgini/metrics.py

@@ -8,7 +8,15 @@ from catboost import CatBoostClassifier, CatBoostRegressor
 from lightgbm import LGBMClassifier, LGBMRegressor
 from numpy import log1p
 from pandas.api.types import is_numeric_dtype
-from sklearn.metrics import get_scorer_names, check_scoring, get_scorer, make_scorer
+from sklearn.metrics import check_scoring, get_scorer, make_scorer
+
+try:
+    from sklearn.metrics import get_scorer_names
+    available_scorers = get_scorer_names()
+except ImportError:
+    from sklearn.metrics._scorer import SCORERS
+    available_scorers = SCORERS
+
 from sklearn.metrics._regression import (
     _check_reg_targets,
     check_consistent_length,
@@ -385,7 +393,6 @@ def _get_scorer(target_type: ModelTaskType, scoring: Union[Callable, str, None])
 
     multiplier = 1
     if isinstance(scoring, str):
-        available_scorers = get_scorer_names()
         metric_name = scoring
         if "mean_squared_log_error" == metric_name or "MSLE" == metric_name or "msle" == metric_name:
             scoring = make_scorer(_ext_mean_squared_log_error, greater_is_better=False)

{upgini-1.1.166a2 → upgini-1.1.168}/src/upgini/resource_bundle/strings.properties

@@ -59,7 +59,7 @@ no_connection_to_upgini=No connection to Upgini server https://search.upgini.com
 no_internet_connection=No internet connection from Jupyter server {} to initiate external data search with Upgini service, please try with Google Colab https://colab.research.google.com
 access_denied=Access denied
 unsupported_search_key=Search key {} not supported
-too_many_generate_features=Too many columns passed in `generate_features` argument. Only two columns supported to generate features now
+too_many_generate_features=Too many columns passed in `generate_features` argument. Only {} columns supported to generate features now
 invalid_round_embeddings=Argument `round_embeddings` should be non negative integer
 no_important_features_for_transform=There are no important features for transform. Return input as transformed
 

{upgini-1.1.166a2 → upgini-1.1.168/src/upgini.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: upgini
-Version: 1.1.166a2
+Version: 1.1.168
 Summary: Intelligent data search & enrichment for Machine Learning
 Home-page: https://upgini.com/
 Author: Upgini Developers