upgini 1.2.74__py3-none-any.whl → 1.2.74a3818.dev1__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.74"
+__version__ = "1.2.74a3818.dev1"

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/autofe/date.py

@@ -187,21 +187,16 @@ class DateListDiff(PandasOperator, DateDiffMixin, ParametrizedOperator):
 class DateListDiffBounded(DateListDiff, ParametrizedOperator):
     lower_bound: Optional[int] = None
     upper_bound: Optional[int] = None
-    normalize: Optional[bool] = None
 
     def to_formula(self) -> str:
         lower_bound = "minusinf" if self.lower_bound is None else self.lower_bound
         upper_bound = "plusinf" if self.upper_bound is None else self.upper_bound
-        norm = "_norm" if self.normalize else ""
-        return f"date_diff_{self.diff_unit}_{lower_bound}_{upper_bound}_{self.aggregation}{norm}"
+        return f"date_diff_{self.diff_unit}_{lower_bound}_{upper_bound}_{self.aggregation}"
 
     @classmethod
     def from_formula(cls, formula: str) -> Optional["DateListDiffBounded"]:
         import re
 
-        normalize = formula.endswith("_norm")
-        formula = formula.replace("_norm", "")
-
         pattern = r"^date_diff_([^_]+)_((minusinf|\d+))_((plusinf|\d+))_(\w+)$"
         match = re.match(pattern, formula)
 
@@ -212,13 +207,8 @@ class DateListDiffBounded(DateListDiff, ParametrizedOperator):
         lower_bound = None if match.group(2) == "minusinf" else int(match.group(2))
         upper_bound = None if match.group(4) == "plusinf" else int(match.group(4))
         aggregation = match.group(6)
-        return cls(
-            diff_unit=diff_unit,
-            lower_bound=lower_bound,
-            upper_bound=upper_bound,
-            aggregation=aggregation,
-            normalize=normalize,
-        )
+
+        return cls(diff_unit=diff_unit, lower_bound=lower_bound, upper_bound=upper_bound, aggregation=aggregation)
 
     def get_params(self) -> Dict[str, Optional[str]]:
         res = super().get_params()
@@ -226,20 +216,14 @@ class DateListDiffBounded(DateListDiff, ParametrizedOperator):
             res["lower_bound"] = str(self.lower_bound)
         if self.upper_bound is not None:
             res["upper_bound"] = str(self.upper_bound)
-        if self.normalize is not None:
-            res["normalize"] = str(self.normalize)
         return res
 
     def _agg(self, x):
-        orig_len = len(x)
         x = x[
             (x >= (self.lower_bound if self.lower_bound is not None else -np.inf))
             & (x < (self.upper_bound if self.upper_bound is not None else np.inf))
         ]
-        agg_res = super()._agg(x)
-        if self.normalize and orig_len > 0:
-            return agg_res / orig_len
-        return agg_res
+        return super()._agg(x)
 
 
 class DatePercentileBase(PandasOperator, abc.ABC):

upgini/autofe/feature.py

@@ -154,34 +154,24 @@ class Feature:
         for child in self.children:
             child.delete_data()
 
-    def get_op_display_name(self, use_alias: bool = True) -> str:
-        return (self.op.alias or self.op.to_formula()).lower() if use_alias else self.op.to_formula()
+    def get_op_display_name(self) -> str:
+        return (self.op.alias or self.op.to_formula()).lower()
 
-    def get_display_name(self, cache: bool = True, shorten: bool = False, use_op_alias: bool = True, **kwargs) -> str:
+    def get_display_name(self, cache: bool = True, shorten: bool = False, **kwargs) -> str:
         if self.cached_display_name is not None and cache:
             return self.cached_display_name
 
         should_stack_op = not isinstance(self.children[0], Column) if self.op.is_unary else False
-        components = []
+        prev_name = [self.children[0].get_op_display_name()] if should_stack_op else []
 
         if self.alias:
-            components.extend(["f_autofe", self.alias])
-        elif should_stack_op:
-            components.extend(
-                [
-                    self.children[0].get_display_name(
-                        cache=cache, shorten=shorten, use_op_alias=use_op_alias, **kwargs
-                    ),
-                    self.get_op_display_name(use_alias=use_op_alias),
-                ]
-            )
-        elif shorten and not self.op.is_unary:
-            components.extend(["f_autofe", self.get_op_display_name(use_alias=use_op_alias)])
+            components = ["f_autofe", self.alias]
+        elif shorten and (not self.op.is_unary or should_stack_op):
+            components = ["f_autofe"] + prev_name + [self.get_op_display_name()]
         else:
-            components = ["f_" + "_f_".join(self.get_columns(**kwargs))] + [
-                "autofe",
-                self.get_op_display_name(use_alias=use_op_alias),
-            ]
+            components = (
+                ["f_" + "_f_".join(self.get_columns(**kwargs))] + ["autofe"] + prev_name + [self.get_op_display_name()]
+            )
         components.extend([str(self.display_index)] if self.display_index is not None else [])
         display_name = "_".join(components)
 

upgini/autofe/unary.py

@@ -1,10 +1,8 @@
-import json
-from typing import Dict, List, Optional
+from typing import Dict, Optional
 import numpy as np
 import pandas as pd
 
 from upgini.autofe.operator import PandasOperator, VectorizableMixin
-from upgini.autofe.utils import pydantic_validator
 
 
 class Abs(PandasOperator, VectorizableMixin):
@@ -155,38 +153,3 @@ class Embeddings(PandasOperator):
     is_unary: bool = True
     input_type: Optional[str] = "string"
     output_type: Optional[str] = "vector"
-
-
-class Bin(PandasOperator):
-    name: str = "bin"
-    is_unary: bool = True
-    output_type: Optional[str] = "category"
-    bin_bounds: List[int] = []
-    is_categorical: bool = True
-
-    def calculate_unary(self, data: pd.Series) -> pd.Series:
-        return data.apply(self._bin, bounds=self.bin_bounds).fillna(-1).astype(int).astype("category")
-
-    def _bin(self, f, bounds):
-        if f is None or np.isnan(f):
-            return np.nan
-        hit = np.where(f >= np.array(bounds))[0]
-        if hit.size > 0:
-            return np.max(hit) + 1
-        else:
-            return np.nan
-
-    def get_params(self) -> Dict[str, Optional[str]]:
-        res = super().get_params()
-        res.update(
-            {
-                "bin_bounds": json.dumps(self.bin_bounds),
-            }
-        )
-        return res
-
-    @pydantic_validator("bin_bounds", mode="before")
-    def parse_bin_bounds(cls, value):
-        if isinstance(value, str):
-            return json.loads(value)
-        return value

upgini/autofe/vector.py

@@ -22,12 +22,3 @@ class Sum(PandasOperator, VectorizableMixin):
 
     def calculate_vector(self, data: List[pd.Series]) -> pd.Series:
         return pd.DataFrame(data).T.fillna(0).sum(axis=1)
-
-
-class Vectorize(PandasOperator, VectorizableMixin):
-    name: str = "vectorize"
-    is_vector: bool = True
-    group_index: int = 0
-
-    def calculate_vector(self, data: List[pd.Series]) -> pd.Series:
-        return pd.DataFrame(data).T.apply(lambda x: x.to_list(), axis=1)

upgini/metrics.py

@@ -326,7 +326,7 @@ class EstimatorWrapper:
         for c in x.columns:
             if is_numeric_dtype(x[c]):
                 x[c] = x[c].astype(float)
-            elif not x[c].dtype == "category":
+            else:
                 x[c] = x[c].astype(str)
 
         if not isinstance(y, pd.Series):

{upgini-1.2.74.dist-info → upgini-1.2.74a3818.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.74
+Version: 1.2.74a3818.dev1
 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.74.dist-info → upgini-1.2.74a3818.dev1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-upgini/__about__.py,sha256=LYhBOE9Gp2v3W8-OC57ijtt8O3FDg9mFfSW0EIsbsG4,23
+upgini/__about__.py,sha256=OJAwNDxLBKa-uno0o9LM57LzTEqZIlSuf5VB46GJths,33
 upgini/__init__.py,sha256=LXSfTNU0HnlOkE69VCxkgIKDhWP-JFo_eBQ71OxTr5Y,261
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=aspri7ZAgwkNNUiIgQ1GRXvw8XQii3F4RfNXSrF4wrw,35365
@@ -6,7 +6,7 @@ upgini/errors.py,sha256=2b_Wbo0OYhLUbrZqdLIx5jBnAsiD1Mcenh-VjR4HCTw,950
 upgini/features_enricher.py,sha256=Li1sPihWVkPUPcma8HRbPFwpCqd9V9d2p5zQUgkpdpU,206998
 upgini/http.py,sha256=RvzcShpDXssLs6ycGN8xilkKi8ZV9XGUrrk8bwdUzbw,43607
 upgini/metadata.py,sha256=Yd6iW2f7Wz6vUkg5uvR4xylN16ANnCKVKqAsAkap7p8,12354
-upgini/metrics.py,sha256=7UhKUwO7bskcu9vgTIQoazE2qrOMhRuFyc6IOu1AJ3Y,39053
+upgini/metrics.py,sha256=UNNA3H7wWATq-lTb9BChDdFc14MOYH9FTWY2Te4OU2o,39024
 upgini/search_task.py,sha256=EuCGp0iCWz2fpuJgN6M47aP_CtIi3Oq9zw78w0mkKiU,17595
 upgini/spinner.py,sha256=4iMd-eIe_BnkqFEMIliULTbj6rNI2HkN_VJ4qYe0cUc,1118
 upgini/version_validator.py,sha256=DvbaAvuYFoJqYt0fitpsk6Xcv-H1BYDJYHUMxaKSH_Y,1509
@@ -14,14 +14,14 @@ upgini/ads_management/__init__.py,sha256=qzyisOToVRP-tquAJD1PblZhNtMrOB8FiyF9Jvf
 upgini/ads_management/ads_manager.py,sha256=igVbN2jz80Umb2BUJixmJVj-zx8unoKpecVo-R-nGdw,2648
 upgini/autofe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 upgini/autofe/all_operators.py,sha256=rdjF5eaE4bC6Q4eu_el5Z7ekYt8DjOFermz2bePPbUc,333
-upgini/autofe/binary.py,sha256=MnQuFiERpocjCPQUjOljlsq5FE-04GPfwtNjzvfNMyU,7671
-upgini/autofe/date.py,sha256=MM1S-6imNSzCDOhbNnmsc_bwSqUWBcS8vWAdHF8j1kY,11134
-upgini/autofe/feature.py,sha256=G_YgnsauIoaMgByx9JXDPiKc4nqs0pwWZUfvoIGMKxY,15305
+upgini/autofe/binary.py,sha256=oOEECc4nRzZN2tYaiqx8F2XHnfWpk1bVvb7ZkZJ0lO8,7709
+upgini/autofe/date.py,sha256=C86F7sPiscUGq2a45UtQA9ADWBWg0kt54mePHHzjbLE,10633
+upgini/autofe/feature.py,sha256=y1x3wijhTVBmloayQAHiscqKU9Ll8kLcGm1PdvS357I,14910
 upgini/autofe/groupby.py,sha256=IYmQV9uoCdRcpkeWZj_kI3ObzoNCNx3ff3h8sTL01tk,3603
 upgini/autofe/operator.py,sha256=EOffJw6vKXpEh5yymqb1RFNJPxGxmnHdFRo9dB5SCFo,4969
-upgini/autofe/unary.py,sha256=Sx11IoHRh5nwyALzjgG9GQOrVNIs8NZ1JzunAJuN66A,5731
+upgini/autofe/unary.py,sha256=yVgPvtfnPSOhrii0YgezddmgWPwyOBCR0JutaIkdTTc,4658
 upgini/autofe/utils.py,sha256=fK1am2_tQj3fL2vDslblye8lmyfWgGIUOX1beYVBz4k,2420
-upgini/autofe/vector.py,sha256=zehv1J9ChHdZKWjKlkRf6RpfQMCJduZmqCEePYNUfkQ,943
+upgini/autofe/vector.py,sha256=l0KdKg-txlZxDSE4hPPfCtfGQofYbl7oaABPr830sPI,667
 upgini/autofe/timeseries/__init__.py,sha256=PGwwDAMwvkXl3el12tXVEmZUgDUvlmIPlXtROm6bD18,738
 upgini/autofe/timeseries/base.py,sha256=rWJqRuFAzTZEsUdWG5s1Vhif9zzRRmalASXvarufRxI,3610
 upgini/autofe/timeseries/cross.py,sha256=BTINVwuZSbm_4NKkVm0FGM68SrvZLENZKXN7-UyvhYI,5319
@@ -70,7 +70,7 @@ upgini/utils/target_utils.py,sha256=P0cCVRaakWLydYwFjk3TEaQfr0p0hfsJCvKRD8qcxiE,
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/ts_utils.py,sha256=26vhC0pN7vLXK6R09EEkMK3Lwb9IVPH7LRdqFIQ3kPs,1383
 upgini/utils/warning_counter.py,sha256=-GRY8EUggEBKODPSuXAkHn9KnEQwAORC0mmz_tim-PM,254
-upgini-1.2.74.dist-info/METADATA,sha256=NK_V10Btl-ASVNrgJZQKOukyMNxod7dz2q_ZwPywTbQ,49091
-upgini-1.2.74.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-upgini-1.2.74.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.74.dist-info/RECORD,,
+upgini-1.2.74a3818.dev1.dist-info/METADATA,sha256=J-UotQ8AHXGCacs9yJb2tQ0s_1Tm89spF2wZgUyysL4,49101
+upgini-1.2.74a3818.dev1.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+upgini-1.2.74a3818.dev1.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.74a3818.dev1.dist-info/RECORD,,