upgini 1.2.0a1__py3-none-any.whl → 1.2.1__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.0a1"
+__version__ = "1.2.1"

upgini/autofe/unary.py

@@ -1,7 +1,6 @@
-from typing import Optional
+from typing import Dict, Optional
 import numpy as np
 import pandas as pd
-from sklearn.preprocessing import Normalizer
 
 from upgini.autofe.operand import PandasOperand, VectorizableMixin
 
@@ -119,17 +118,32 @@ class Norm(PandasOperand):
     name: str = "norm"
     is_unary: bool = True
     output_type: Optional[str] = "float"
+    norm: Optional[float] = None
 
     def calculate_unary(self, data: pd.Series) -> pd.Series:
         data_dropna = data.dropna()
         if data_dropna.empty:
             return data
 
-        normalized_data = Normalizer().transform(data_dropna.to_frame().T).T
-        normalized_data = pd.Series(normalized_data[:, 0], index=data_dropna.index, name=data.name)
-        normalized_data = normalized_data.reindex(data.index)
+        if self.norm is not None:
+            normalized_data = data / self.norm
+        else:
+            self.norm = np.sqrt(np.sum(data * data))
+            normalized_data = data / self.norm
+
         return normalized_data
 
+    def set_params(self, params: Dict[str, str]):
+        super().set_params(params)
+        if "norm" in params:
+            self.norm = params["norm"]
+        return self
+
+    def get_params(self) -> Dict[str, Optional[str]]:
+        res = super().get_params()
+        res["norm"] = self.norm
+        return res
+
 
 class Embeddings(PandasOperand):
     name: str = "emb"

{upgini-1.2.0a1.dist-info → upgini-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.0a1
+Version: 1.2.1
 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.0a1.dist-info → upgini-1.2.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-upgini/__about__.py,sha256=dMk28IuEJr_qWW7xH2uH2BnZ8G_djCORGGd6opmGetw,24
+upgini/__about__.py,sha256=Mlm4Gvmb_6yQxwUbv2Ksc-BJFXLPg9H1Vt2iV7wXrA4,22
 upgini/__init__.py,sha256=Xs0YFVBu1KUdtZzbStGRPQtLt3YLzJnjx5nIUBlX8BE,415
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=olZ-OHSfBNoBSCo7R5t7uCLukI2nO7afpx_A-HCiJLk,31067
@@ -20,7 +20,7 @@ upgini/autofe/date.py,sha256=OpFc3Al0xO3qlESn2Uokfxw51ArVqmh3xngWwdrsaqE,9762
 upgini/autofe/feature.py,sha256=gwGWY2UcX_0wHAvfEiu1rRU7GFZyzMWZIaPVcf6kD80,14223
 upgini/autofe/groupby.py,sha256=r-xl_keZZgm_tpiEoDhjYSkT6NHv7a4cRQR4wJ4uCp8,3263
 upgini/autofe/operand.py,sha256=uk883RaNqgXqtkaRqA1re1d9OFnnpv0JVvelYx09Yw0,2943
-upgini/autofe/unary.py,sha256=RiK-Fz3fgjPlqWWfro6x7qChjEZ8W8RTnl5-MT1kaQA,4218
+upgini/autofe/unary.py,sha256=VTX5BdPJUArt-H9qYfOghKw_WlUatFjZto6zu3KGjb4,4484
 upgini/autofe/vector.py,sha256=ehcZUDqV71TfbU8EmKfdYp603gS2dJY_-fpr10ho5sI,663
 upgini/data_source/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 upgini/data_source/data_source_publisher.py,sha256=Vg0biG86YB0OEaoxbK9YYrr4yARm11_h3bTWIBgoScA,22115
@@ -57,7 +57,7 @@ upgini/utils/sklearn_ext.py,sha256=13jQS_k7v0aUtudXV6nGUEWjttPQzAW9AFYL5wgEz9k,4
 upgini/utils/target_utils.py,sha256=BVtDmrmFMKerSUWaNOIEdzsYHIFiODdpnWbE50QDPDc,7864
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/warning_counter.py,sha256=dIWBB4dI5XRRJZudvIlqlIYKEiwLLPcXarsZuYRt338,227
-upgini-1.2.0a1.dist-info/METADATA,sha256=568JisotupYzFolx0QDyv_qN5CtSIEuHuium23_SDp8,48230
-upgini-1.2.0a1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-upgini-1.2.0a1.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.0a1.dist-info/RECORD,,
+upgini-1.2.1.dist-info/METADATA,sha256=6kMBYOGEY3dcShtSVLc1Qo9kyL8dKwdmFXZXhM1vzFA,48228
+upgini-1.2.1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.2.1.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.1.dist-info/RECORD,,