upgini 1.2.78__py3-none-any.whl → 1.2.79__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.78"
+__version__ = "1.2.79"

upgini/autofe/vector.py

@@ -1,8 +1,8 @@
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 import pandas as pd
 
-from upgini.autofe.operator import PandasOperator, VectorizableMixin
+from upgini.autofe.operator import OperatorRegistry, PandasOperator, VectorizableMixin
 
 
 class Mean(PandasOperator, VectorizableMixin):
@@ -31,3 +31,19 @@ class Vectorize(PandasOperator, VectorizableMixin):
 
     def calculate_vector(self, data: List[pd.Series]) -> pd.Series:
         return pd.DataFrame(data).T.apply(lambda x: x.to_list(), 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

upgini/metrics.py

@@ -87,7 +87,6 @@ CATBOOST_MULTICLASS_PARAMS = {
 
 LIGHTGBM_REGRESSION_PARAMS = {
     "random_state": DEFAULT_RANDOM_STATE,
-    "deterministic": True,
     "min_gain_to_split": 0.001,
     "n_estimators": 275,
     "max_depth": 5,
@@ -99,6 +98,8 @@ LIGHTGBM_REGRESSION_PARAMS = {
     "feature_fraction": 1.0,
     "min_sum_hessian_in_leaf": 0.01,
     "objective": "huber",
+    "deterministic": "true",
+    "force_col_wise": "true",
     "verbosity": -1,
 }
 
@@ -117,6 +118,8 @@ LIGHTGBM_MULTICLASS_PARAMS = {
     "use_quantized_grad": "true",
     "num_grad_quant_bins": "8",
     "stochastic_rounding": "true",
+    "deterministic": "true",
+    "force_col_wise": "true",
     "verbosity": -1,
 }
 
@@ -128,11 +131,12 @@ LIGHTGBM_BINARY_PARAMS = {
     "learning_rate": 0.05,
     "objective": "binary",
     # "class_weight": "balanced",
-    "deterministic": True,
     "max_cat_threshold": 80,
     "min_data_per_group": 20,
     "cat_smooth": 18,
     "cat_l2": 8,
+    "deterministic": "true",
+    "force_col_wise": "true",
     "verbosity": -1,
 }
 

{upgini-1.2.78.dist-info → upgini-1.2.79.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.78
+Version: 1.2.79
 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.78.dist-info → upgini-1.2.79.dist-info}/RECORD

@@ -1,4 +1,4 @@
-upgini/__about__.py,sha256=um-GnDBOmvXVFDzFeMIfwSF9jG0V0HqCNhYlIkfUoOQ,23
+upgini/__about__.py,sha256=mupwAhPLfGDd9OAn7f6J2lwQapeaIysxn41khUOG57I,23
 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=_UkJS35uGaYtI7dR6Xd9Q28nmiPzTjhK3y8v3IjJTfQ,208245
 upgini/http.py,sha256=UH7nswcZ221un3O_VW9limCBO5oRsyg1eKUHiVslRPs,43737
 upgini/metadata.py,sha256=Yd6iW2f7Wz6vUkg5uvR4xylN16ANnCKVKqAsAkap7p8,12354
-upgini/metrics.py,sha256=1AgIPEzd4sGJ8AlMrJziQruT-M1-lGgpnDIXUYeQJ6Q,39171
+upgini/metrics.py,sha256=_Ue1nymBMVmoCdPMGGXe0FFkvNxNg592FmX2WJWKDFE,39294
 upgini/search_task.py,sha256=RcvAE785yksWTsTNWuZFVNlk32jHElMoEna1T_C5N8Q,17823
 upgini/spinner.py,sha256=4iMd-eIe_BnkqFEMIliULTbj6rNI2HkN_VJ4qYe0cUc,1118
 upgini/version_validator.py,sha256=DvbaAvuYFoJqYt0fitpsk6Xcv-H1BYDJYHUMxaKSH_Y,1509
@@ -21,7 +21,7 @@ upgini/autofe/groupby.py,sha256=IYmQV9uoCdRcpkeWZj_kI3ObzoNCNx3ff3h8sTL01tk,3603
 upgini/autofe/operator.py,sha256=EOffJw6vKXpEh5yymqb1RFNJPxGxmnHdFRo9dB5SCFo,4969
 upgini/autofe/unary.py,sha256=Sx11IoHRh5nwyALzjgG9GQOrVNIs8NZ1JzunAJuN66A,5731
 upgini/autofe/utils.py,sha256=fK1am2_tQj3fL2vDslblye8lmyfWgGIUOX1beYVBz4k,2420
-upgini/autofe/vector.py,sha256=zehv1J9ChHdZKWjKlkRf6RpfQMCJduZmqCEePYNUfkQ,943
+upgini/autofe/vector.py,sha256=jHs0nNTOaHspYUlxW7fjQepk4cvr_JDQ65L1OCiVsds,1360
 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.78.dist-info/METADATA,sha256=9NDKzhMak5vKoLrxoAAfTjU-HsTJByudWztn1gxxJyA,49091
-upgini-1.2.78.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-upgini-1.2.78.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.78.dist-info/RECORD,,
+upgini-1.2.79.dist-info/METADATA,sha256=zxQyk76bWj5AGiXERdfEQVcqsZXD4RU5gaRQGk1LpeM,49091
+upgini-1.2.79.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.2.79.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.79.dist-info/RECORD,,