upgini 1.2.71a3832.dev7__py3-none-any.whl → 1.2.71a3832.dev8__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.
upgini/__about__.py CHANGED
@@ -1 +1 @@
1
- __version__ = "1.2.71a3832.dev7"
1
+ __version__ = "1.2.71a3832.dev8"
upgini/metrics.py CHANGED
@@ -765,6 +765,8 @@ class LightGBMWrapper(EstimatorWrapper):
765
765
  if LIGHTGBM_EARLY_STOPPING_ROUNDS is not None:
766
766
  params["callbacks"] = [lgb.early_stopping(stopping_rounds=LIGHTGBM_EARLY_STOPPING_ROUNDS, verbose=False)]
767
767
  self.cat_features = _get_cat_features(x)
768
+ if self.cat_features:
769
+ params["categorical_feature"] = self.cat_features
768
770
  x = fill_na_cat_features(x, self.cat_features)
769
771
  for feature in self.cat_features:
770
772
  x[feature] = x[feature].astype("category").cat.codes
@@ -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) and "callbacks" not in fit_params:
518
+ elif is_lightgbm_estimator(estimator):
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,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: upgini
3
- Version: 1.2.71a3832.dev7
3
+ Version: 1.2.71a3832.dev8
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/
@@ -1,4 +1,4 @@
1
- upgini/__about__.py,sha256=OM1SdEE1I5fkWcKRuW6LMCMJ-zfyOStMB1G7AONAIvI,33
1
+ upgini/__about__.py,sha256=dqxmBK1lI76uxRYzaLNpLnZdX9JQ6vbLBh1yKTLqMsg,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=cY9exGDlpmNcdvblhPVwBmBSlD_JyrgzMUMgh2r9r08,38440
9
+ upgini/metrics.py,sha256=J_WX81v-XI61HybTbEKcO17grr1qM8BOXCqQwBD4OcQ,38532
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=w27qbgzp5H-4W0LOHHzPUjtcVSchvJHw5sv5BFxNkt8,45076
67
+ upgini/utils/sklearn_ext.py,sha256=HpaNQaKJisgNE7IZ71n7uswxTj7kbPglU2G3s1sORAc,45042
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.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,,
73
+ upgini-1.2.71a3832.dev8.dist-info/METADATA,sha256=Fihaz0CDiEJbFoqb3tIH8rm6ft5Awa2-P_22m2rJqnI,49101
74
+ upgini-1.2.71a3832.dev8.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
75
+ upgini-1.2.71a3832.dev8.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
76
+ upgini-1.2.71a3832.dev8.dist-info/RECORD,,