upgini 1.2.71a3832.dev3__tar.gz → 1.2.71a3832.dev4__tar.gz

{upgini-1.2.71a3832.dev3 → upgini-1.2.71a3832.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.71a3832.dev3
+Version: 1.2.71a3832.dev4
 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.71a3832.dev4/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.71a3832.dev4"

{upgini-1.2.71a3832.dev3 → upgini-1.2.71a3832.dev4}/src/upgini/metrics.py

@@ -12,6 +12,7 @@ from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 import numpy as np
 import pandas as pd
 from lightgbm import LGBMClassifier, LGBMRegressor
+import lightgbm as lgb
 from numpy import log1p
 from pandas.api.types import is_numeric_dtype
 from sklearn.metrics import check_scoring, get_scorer, make_scorer, roc_auc_score
@@ -759,6 +760,7 @@ class LightGBMWrapper(EstimatorWrapper):
 
     def _prepare_to_fit(self, x: pd.DataFrame, y: pd.Series) -> Tuple[pd.DataFrame, pd.Series, np.ndarray, dict]:
         x, y_numpy, groups, params = super()._prepare_to_fit(x, y)
+        params["callbacks"] = [lgb.early_stopping(stopping_rounds=20)]
         self.cat_features = _get_cat_features(x)
         x = fill_na_cat_features(x, self.cat_features)
         for feature in self.cat_features:

{upgini-1.2.71a3832.dev3 → upgini-1.2.71a3832.dev4}/src/upgini/utils/sklearn_ext.py

@@ -349,6 +349,14 @@ def is_catboost_estimator(estimator):
         return False
 
 
+def is_lightgbm_estimator(estimator):
+    try:
+        from lightgbm import LGBMClassifier, LGBMRegressor
+        return isinstance(estimator, (LGBMClassifier, LGBMRegressor))
+    except ImportError:
+        return False
+
+
 def _fit_and_score(
     estimator,
     X,
@@ -507,6 +515,9 @@ def _fit_and_score(
             if is_catboost_estimator(estimator):
                 fit_params = fit_params.copy()
                 fit_params["eval_set"] = [(X_test, y_test)]
+            elif is_lightgbm_estimator(estimator):
+                fit_params = fit_params.copy()
+                fit_params["eval_set"] = [(X_test, y_test)]
             estimator.fit(X_train, y_train, **fit_params)
 
     except Exception:

upgini-1.2.71a3832.dev3/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.71a3832.dev3"

upgini-1.2.71a3832.dev3/src/upgini/lazy_import.py

@@ -1,35 +0,0 @@
-import importlib
-import importlib.util
-import importlib.machinery
-
-
-class LazyImport:
-    def __init__(self, module_name, class_name):
-        self.module_name = module_name
-        self.class_name = class_name
-        self._module = None
-        self._class = None
-
-    def _load(self):
-        if self._module is None:
-            # Load module and save link to it
-            spec = importlib.util.find_spec(self.module_name)
-            if spec is None:
-                raise ImportError(f"Module {self.module_name} not found")
-
-            # Create module
-            self._module = importlib.util.module_from_spec(spec)
-
-            # Execute module
-            spec.loader.exec_module(self._module)
-
-            # Get class from module
-            self._class = getattr(self._module, self.class_name)
-
-    def __call__(self, *args, **kwargs):
-        self._load()
-        return self._class(*args, **kwargs)
-
-    def __getattr__(self, name):
-        self._load()
-        return getattr(self._class, name)