upgini 1.2.71a3832.dev5__py3-none-any.whl → 1.2.71a3832.dev7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. upgini/__about__.py +1 -1
  2. upgini/metrics.py +4 -1
  3. upgini/utils/sklearn_ext.py +1 -1
  4. {upgini-1.2.71a3832.dev5.dist-info → upgini-1.2.71a3832.dev7.dist-info}/METADATA +1 -1
  5. {upgini-1.2.71a3832.dev5.dist-info → upgini-1.2.71a3832.dev7.dist-info}/RECORD +7 -7
  6. {upgini-1.2.71a3832.dev5.dist-info → upgini-1.2.71a3832.dev7.dist-info}/WHEEL +0 -0
  7. {upgini-1.2.71a3832.dev5.dist-info → upgini-1.2.71a3832.dev7.dist-info}/licenses/LICENSE +0 -0
upgini/__about__.py CHANGED
@@ -1 +1 @@
1
- __version__ = "1.2.71a3832.dev5"
1
+ __version__ = "1.2.71a3832.dev7"
upgini/metrics.py CHANGED
@@ -155,6 +155,8 @@ LIGHTGBM_BINARY_PARAMS = {
155
155
  "verbosity": -1,
156
156
  }
157
157
 
158
+ LIGHTGBM_EARLY_STOPPING_ROUNDS = 20
159
+
158
160
  N_FOLDS = 5
159
161
  BLOCKED_TS_TEST_SIZE = 0.2
160
162
 
@@ -760,7 +762,8 @@ class LightGBMWrapper(EstimatorWrapper):
760
762
 
761
763
  def _prepare_to_fit(self, x: pd.DataFrame, y: pd.Series) -> Tuple[pd.DataFrame, pd.Series, np.ndarray, dict]:
762
764
  x, y_numpy, groups, params = super()._prepare_to_fit(x, y)
763
- params["callbacks"] = [lgb.early_stopping(stopping_rounds=20, verbose=False)]
765
+ if LIGHTGBM_EARLY_STOPPING_ROUNDS is not None:
766
+ params["callbacks"] = [lgb.early_stopping(stopping_rounds=LIGHTGBM_EARLY_STOPPING_ROUNDS, verbose=False)]
764
767
  self.cat_features = _get_cat_features(x)
765
768
  x = fill_na_cat_features(x, self.cat_features)
766
769
  for feature in self.cat_features:
upgini/utils/sklearn_ext.py CHANGED
@@ -515,7 +515,7 @@ def _fit_and_score(
515
515
  if is_catboost_estimator(estimator):
516
516
  fit_params = fit_params.copy()
517
517
  fit_params["eval_set"] = [(X_test, y_test)]
518
- elif is_lightgbm_estimator(estimator):
518
+ elif is_lightgbm_estimator(estimator) and "callbacks" not in fit_params:
519
519
  fit_params = fit_params.copy()
520
520
  fit_params["eval_set"] = [(X_test, y_test)]
521
521
  estimator.fit(X_train, y_train, **fit_params)
{upgini-1.2.71a3832.dev5.dist-info → upgini-1.2.71a3832.dev7.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: upgini
3
- Version: 1.2.71a3832.dev5
3
+ Version: 1.2.71a3832.dev7
4
4
  Summary: Intelligent data search & enrichment for Machine Learning
5
5
  Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
6
6
  Project-URL: Homepage, https://upgini.com/
{upgini-1.2.71a3832.dev5.dist-info → upgini-1.2.71a3832.dev7.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- upgini/__about__.py,sha256=60wCW3ThZS1gzPZ6xysuJt1G6Z3eD7XVIfHvTsVsOVI,33
1
+ upgini/__about__.py,sha256=OM1SdEE1I5fkWcKRuW6LMCMJ-zfyOStMB1G7AONAIvI,33
2
2
  upgini/__init__.py,sha256=LXSfTNU0HnlOkE69VCxkgIKDhWP-JFo_eBQ71OxTr5Y,261
3
3
  upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
4
4
  upgini/dataset.py,sha256=aspri7ZAgwkNNUiIgQ1GRXvw8XQii3F4RfNXSrF4wrw,35365
@@ -6,7 +6,7 @@ upgini/errors.py,sha256=2b_Wbo0OYhLUbrZqdLIx5jBnAsiD1Mcenh-VjR4HCTw,950
6
6
  upgini/features_enricher.py,sha256=Z6RSjqcqneGwWflsq1Q5rjf83awPNYqKpAgHRh7jils,204680
7
7
  upgini/http.py,sha256=RvzcShpDXssLs6ycGN8xilkKi8ZV9XGUrrk8bwdUzbw,43607
8
8
  upgini/metadata.py,sha256=Yd6iW2f7Wz6vUkg5uvR4xylN16ANnCKVKqAsAkap7p8,12354
9
- upgini/metrics.py,sha256=zVUWBeph9ESfhrLTIzjFnXU2M-1m6TIZy8HKyBSOpfY,38316
9
+ upgini/metrics.py,sha256=cY9exGDlpmNcdvblhPVwBmBSlD_JyrgzMUMgh2r9r08,38440
10
10
  upgini/search_task.py,sha256=EuCGp0iCWz2fpuJgN6M47aP_CtIi3Oq9zw78w0mkKiU,17595
11
11
  upgini/spinner.py,sha256=4iMd-eIe_BnkqFEMIliULTbj6rNI2HkN_VJ4qYe0cUc,1118
12
12
  upgini/version_validator.py,sha256=DvbaAvuYFoJqYt0fitpsk6Xcv-H1BYDJYHUMxaKSH_Y,1509
@@ -64,13 +64,13 @@ upgini/utils/mstats.py,sha256=u3gQVUtDRbyrOQK6V1UJ2Rx1QbkSNYGjXa6m3Z_dPVs,6286
64
64
  upgini/utils/phone_utils.py,sha256=IrbztLuOJBiePqqxllfABWfYlfAjYevPhXKipl95wUI,10432
65
65
  upgini/utils/postal_code_utils.py,sha256=5M0sUqH2DAr33kARWCTXR-ACyzWbjDq_-0mmEml6ZcU,1716
66
66
  upgini/utils/progress_bar.py,sha256=N-Sfdah2Hg8lXP_fV9EfUTXz_PyRt4lo9fAHoUDOoLc,1550
67
- upgini/utils/sklearn_ext.py,sha256=HpaNQaKJisgNE7IZ71n7uswxTj7kbPglU2G3s1sORAc,45042
67
+ upgini/utils/sklearn_ext.py,sha256=w27qbgzp5H-4W0LOHHzPUjtcVSchvJHw5sv5BFxNkt8,45076
68
68
  upgini/utils/sort.py,sha256=8uuHs2nfSMVnz8GgvbOmgMB1PgEIZP1uhmeRFxcwnYw,7039
69
69
  upgini/utils/target_utils.py,sha256=b1GzO8_gMcwXSZ2v98CY50MJJBzKbWHId_BJGybXfkM,16579
70
70
  upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
71
71
  upgini/utils/ts_utils.py,sha256=26vhC0pN7vLXK6R09EEkMK3Lwb9IVPH7LRdqFIQ3kPs,1383
72
72
  upgini/utils/warning_counter.py,sha256=-GRY8EUggEBKODPSuXAkHn9KnEQwAORC0mmz_tim-PM,254
73
- upgini-1.2.71a3832.dev5.dist-info/METADATA,sha256=1u01jLu_RPP6lvkPMLDd45E1RS1ikaaQgilTxjekhjs,49101
74
- upgini-1.2.71a3832.dev5.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
75
- upgini-1.2.71a3832.dev5.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
76
- upgini-1.2.71a3832.dev5.dist-info/RECORD,,
73
+ upgini-1.2.71a3832.dev7.dist-info/METADATA,sha256=BqO8gwjY5CjKw0UkrCOczYJQfbwyKlarV3WYCMbEBog,49101
74
+ upgini-1.2.71a3832.dev7.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
75
+ upgini-1.2.71a3832.dev7.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
76
+ upgini-1.2.71a3832.dev7.dist-info/RECORD,,