upgini 1.2.81a3832.dev14__tar.gz → 1.2.81a3832.dev16__tar.gz

{upgini-1.2.81a3832.dev14 → upgini-1.2.81a3832.dev16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.81a3832.dev14
+Version: 1.2.81a3832.dev16
 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.81a3832.dev16/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.81a3832.dev16"

{upgini-1.2.81a3832.dev14 → upgini-1.2.81a3832.dev16}/src/upgini/autofe/binary.py

@@ -146,8 +146,8 @@ class Distance(PandasOperator):
 
     # row-wise dot product, handling None values
     def __dot(self, left: pd.Series, right: pd.Series) -> pd.Series:
-        left = left.apply(lambda x: np.array(x))
-        right = right.apply(lambda x: np.array(x))
+        left = left.apply(lambda x: np.array(x).astype(np.float64))
+        right = right.apply(lambda x: np.array(x).astype(np.float64))
         res = (left.dropna() * right.dropna()).apply(np.sum)
         res = res.reindex(left.index.union(right.index))
         return res

{upgini-1.2.81a3832.dev14 → upgini-1.2.81a3832.dev16}/src/upgini/features_enricher.py

@@ -3937,7 +3937,10 @@ if response.status_code == 200:
             if feature_meta.name in original_names_dict.keys():
                 feature_meta.name = original_names_dict[feature_meta.name]
 
-            if updated_shaps is not None:
+            is_client_feature = feature_meta.name in df.columns
+
+            # Show and update shap values for client features only if select_features is True
+            if updated_shaps is not None and (not is_client_feature or self.fit_select_features):
                 updating_shap = updated_shaps.get(feature_meta.name)
                 if updating_shap is None:
                     if feature_meta.shap_value != 0.0:

{upgini-1.2.81a3832.dev14 → upgini-1.2.81a3832.dev16}/src/upgini/metadata.py

@@ -326,7 +326,7 @@ class RuntimeParameters(BaseModel):
 
 
 class AutoFEParameters(BaseModel):
-    ts_gap_days: Optional[int] = None
+    ts_gap_days: int = 0
 
 
 class SearchCustomization(BaseModel):

{upgini-1.2.81a3832.dev14 → upgini-1.2.81a3832.dev16}/src/upgini/metrics.py

@@ -91,17 +91,8 @@ CATBOOST_MULTICLASS_PARAMS = {
 
 LIGHTGBM_REGRESSION_PARAMS = {
     "random_state": DEFAULT_RANDOM_STATE,
-    "min_gain_to_split": 0.001,
     "n_estimators": 275,
-    "max_depth": 5,
-    "max_cat_threshold": 80,
-    "min_data_per_group": 25,
-    "cat_l2": 10,
-    "cat_smooth": 12,
-    "learning_rate": 0.05,
     "feature_fraction": 1.0,
-    "min_sum_hessian_in_leaf": 0.01,
-    "objective": "huber",
     "deterministic": "true",
     "verbosity": -1,
 }

upgini-1.2.81a3832.dev14/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.81a3832.dev14"