upgini 1.2.71a3832.dev4__tar.gz → 1.2.71a3832.dev6__tar.gz

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 (78) hide show
  1. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/PKG-INFO +1 -1
  2. upgini-1.2.71a3832.dev6/src/upgini/__about__.py +1 -0
  3. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/metrics.py +3 -1
  4. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/sklearn_ext.py +1 -1
  5. upgini-1.2.71a3832.dev4/src/upgini/__about__.py +0 -1
  6. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/.gitignore +0 -0
  7. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/LICENSE +0 -0
  8. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/README.md +0 -0
  9. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/pyproject.toml +0 -0
  10. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/__init__.py +0 -0
  11. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/ads.py +0 -0
  12. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/ads_management/__init__.py +0 -0
  13. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/ads_management/ads_manager.py +0 -0
  14. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/__init__.py +0 -0
  15. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/all_operators.py +0 -0
  16. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/binary.py +0 -0
  17. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/date.py +0 -0
  18. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/feature.py +0 -0
  19. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/groupby.py +0 -0
  20. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/operator.py +0 -0
  21. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/timeseries/__init__.py +0 -0
  22. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/timeseries/base.py +0 -0
  23. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/timeseries/cross.py +0 -0
  24. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/timeseries/delta.py +0 -0
  25. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/timeseries/lag.py +0 -0
  26. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/timeseries/roll.py +0 -0
  27. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/timeseries/trend.py +0 -0
  28. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/timeseries/volatility.py +0 -0
  29. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/unary.py +0 -0
  30. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/utils.py +0 -0
  31. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/autofe/vector.py +0 -0
  32. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/data_source/__init__.py +0 -0
  33. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/data_source/data_source_publisher.py +0 -0
  34. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/dataset.py +0 -0
  35. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/errors.py +0 -0
  36. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/features_enricher.py +0 -0
  37. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/http.py +0 -0
  38. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/mdc/__init__.py +0 -0
  39. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/mdc/context.py +0 -0
  40. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/metadata.py +0 -0
  41. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/normalizer/__init__.py +0 -0
  42. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/normalizer/normalize_utils.py +0 -0
  43. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/resource_bundle/__init__.py +0 -0
  44. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/resource_bundle/exceptions.py +0 -0
  45. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/resource_bundle/strings.properties +0 -0
  46. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/resource_bundle/strings_widget.properties +0 -0
  47. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/sampler/__init__.py +0 -0
  48. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/sampler/base.py +0 -0
  49. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/sampler/random_under_sampler.py +0 -0
  50. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/sampler/utils.py +0 -0
  51. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/search_task.py +0 -0
  52. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/spinner.py +0 -0
  53. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/Roboto-Regular.ttf +0 -0
  54. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/__init__.py +0 -0
  55. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/base_search_key_detector.py +0 -0
  56. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/blocked_time_series.py +0 -0
  57. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/country_utils.py +0 -0
  58. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/custom_loss_utils.py +0 -0
  59. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/cv_utils.py +0 -0
  60. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/datetime_utils.py +0 -0
  61. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/deduplicate_utils.py +0 -0
  62. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/display_utils.py +0 -0
  63. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/email_utils.py +0 -0
  64. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/fallback_progress_bar.py +0 -0
  65. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/feature_info.py +0 -0
  66. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/features_validator.py +0 -0
  67. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/format.py +0 -0
  68. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/ip_utils.py +0 -0
  69. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/mstats.py +0 -0
  70. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/phone_utils.py +0 -0
  71. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/postal_code_utils.py +0 -0
  72. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/progress_bar.py +0 -0
  73. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/sort.py +0 -0
  74. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/target_utils.py +0 -0
  75. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/track_info.py +0 -0
  76. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/ts_utils.py +0 -0
  77. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/utils/warning_counter.py +0 -0
  78. {upgini-1.2.71a3832.dev4 → upgini-1.2.71a3832.dev6}/src/upgini/version_validator.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: upgini
3
- Version: 1.2.71a3832.dev4
3
+ Version: 1.2.71a3832.dev6
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/
@@ -0,0 +1 @@
1
+ __version__ = "1.2.71a3832.dev6"
@@ -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,7 @@ 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)]
765
+ params["callbacks"] = [lgb.early_stopping(stopping_rounds=LIGHTGBM_EARLY_STOPPING_ROUNDS, verbose=False)]
764
766
  self.cat_features = _get_cat_features(x)
765
767
  x = fill_na_cat_features(x, self.cat_features)
766
768
  for feature in self.cat_features:
@@ -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)
@@ -1 +0,0 @@
1
- __version__ = "1.2.71a3832.dev4"