upgini 1.2.72__tar.gz → 1.2.73a3659.dev2__tar.gz

{upgini-1.2.72 → upgini-1.2.73a3659.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.72
+Version: 1.2.73a3659.dev2
 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.73a3659.dev2/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.73a3659.dev2"

upgini-1.2.73a3659.dev2/src/upgini/autofe/vector.py

@@ -0,0 +1,46 @@
+from typing import Dict, List, Optional
+
+import pandas as pd
+
+from upgini.autofe.operator import OperatorRegistry, PandasOperator, VectorizableMixin
+
+
+class Mean(PandasOperator, VectorizableMixin):
+    name: str = "mean"
+    output_type: Optional[str] = "float"
+    is_vector: bool = True
+    group_index: int = 0
+
+    def calculate_vector(self, data: List[pd.Series]) -> pd.Series:
+        return pd.DataFrame(data).T.fillna(0).mean(axis=1)
+
+
+class Sum(PandasOperator, VectorizableMixin):
+    name: str = "sum"
+    is_vector: bool = True
+    group_index: int = 0
+
+    def calculate_vector(self, data: List[pd.Series]) -> pd.Series:
+        return pd.DataFrame(data).T.fillna(0).sum(axis=1)
+
+
+class OnnxModel(PandasOperator, metaclass=OperatorRegistry):
+    name: str = "onnx"
+    is_vector: bool = True
+    output_type: Optional[str] = "float"
+    model_name: str = ""
+
+    def get_params(self) -> Dict[str, Optional[str]]:
+        res = super().get_params()
+        res.update(
+            {
+                "model_name": self.model_name,
+            }
+        )
+        return res
+
+    # def load_model(self):
+    #     ...
+
+    # def calculate_vector(self, data: List[pd.Series]) -> pd.Series:
+    #     ...

upgini-1.2.72/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.72"

upgini-1.2.72/src/upgini/autofe/vector.py

@@ -1,24 +0,0 @@
-from typing import List, Optional
-
-import pandas as pd
-
-from upgini.autofe.operator import PandasOperator, VectorizableMixin
-
-
-class Mean(PandasOperator, VectorizableMixin):
-    name: str = "mean"
-    output_type: Optional[str] = "float"
-    is_vector: bool = True
-    group_index: int = 0
-
-    def calculate_vector(self, data: List[pd.Series]) -> pd.Series:
-        return pd.DataFrame(data).T.fillna(0).mean(axis=1)
-
-
-class Sum(PandasOperator, VectorizableMixin):
-    name: str = "sum"
-    is_vector: bool = True
-    group_index: int = 0
-
-    def calculate_vector(self, data: List[pd.Series]) -> pd.Series:
-        return pd.DataFrame(data).T.fillna(0).sum(axis=1)