upgini 1.2.116__py3-none-any.whl → 1.2.117__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.116"
1
+ __version__ = "1.2.117"
upgini/metrics.py CHANGED
@@ -847,7 +847,7 @@ class CatBoostWrapper(EstimatorWrapper):
847
847
 
848
848
  feature_importance = {}
849
849
  for i, col in enumerate(x.columns):
850
- feature_importance[col] = np.mean(np.abs(shap_values[:, i]))
850
+ feature_importance[col] = float(np.mean(np.abs(shap_values[:, i])))
851
851
 
852
852
  return feature_importance
853
853
 
@@ -922,6 +922,7 @@ class LightGBMWrapper(EstimatorWrapper):
922
922
  encoded = cat_encoder.transform(x_copy[self.cat_features]).astype(int)
923
923
  else:
924
924
  encoded = cat_encoder.transform(x_copy[self.cat_features]).astype("category")
925
+ x_copy = x_copy.drop(columns=self.cat_features, errors="ignore")
925
926
  x_copy[self.cat_features] = encoded
926
927
 
927
928
  shap_matrix = estimator.predict(
@@ -943,7 +944,7 @@ class LightGBMWrapper(EstimatorWrapper):
943
944
 
944
945
  feature_importance = {}
945
946
  for i, col in enumerate(x.columns):
946
- feature_importance[col] = np.mean(np.abs(shap_matrix[:, i]))
947
+ feature_importance[col] = float(np.mean(np.abs(shap_matrix[:, i])))
947
948
 
948
949
  return feature_importance
949
950
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: upgini
3
- Version: 1.2.116
3
+ Version: 1.2.117
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/
@@ -19,9 +19,10 @@ Classifier: License :: OSI Approved :: BSD License
19
19
  Classifier: Operating System :: OS Independent
20
20
  Classifier: Programming Language :: Python :: 3.10
21
21
  Classifier: Programming Language :: Python :: 3.11
22
+ Classifier: Programming Language :: Python :: 3.12
22
23
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
23
24
  Classifier: Topic :: Scientific/Engineering :: Information Analysis
24
- Requires-Python: <3.12,>=3.10
25
+ Requires-Python: <3.13,>=3.10
25
26
  Requires-Dist: catboost>=1.0.3
26
27
  Requires-Dist: category-encoders>=2.8.1
27
28
  Requires-Dist: fastparquet>=0.8.1
@@ -1,4 +1,4 @@
1
- upgini/__about__.py,sha256=uljlkRI1AQACoivbMb5ybeY28dehm84UBPMxgB7YFtM,24
1
+ upgini/__about__.py,sha256=MY7Fho70n94XaciDTteAp4gxVleTVF6idcG3aECjijI,24
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=pQ8JQe0cdygD-W9GefJmfE6bnj4EYzXsjlgWdIS9nS8,31578
@@ -6,7 +6,7 @@ upgini/errors.py,sha256=2b_Wbo0OYhLUbrZqdLIx5jBnAsiD1Mcenh-VjR4HCTw,950
6
6
  upgini/features_enricher.py,sha256=iYay-Ye5WGntieg3X7uyg9W3x_1FUELrmhJnJIvQMeI,228897
7
7
  upgini/http.py,sha256=-J_wOpnwVnT0ebPC6sOs6fN3AWtCD0LJLu6nlYmxaqk,44348
8
8
  upgini/metadata.py,sha256=VzgtgEbPPtNxTrj9LM5qSDP3DujHwAXqbUSKBjPcb9c,12477
9
- upgini/metrics.py,sha256=_kBg6gSXx82LRlRadg-Qggm-GtcPBLbtK3nGCKQjELo,45925
9
+ upgini/metrics.py,sha256=KCPE_apPN-9BIdv6GqASbJVaB_gBcy8wzNApAcyaGo4,46020
10
10
  upgini/search_task.py,sha256=SAiUd1AytbA2Q6PSnnztr7oTRKpud1wQZ5YtKjsmQHU,18256
11
11
  upgini/spinner.py,sha256=4iMd-eIe_BnkqFEMIliULTbj6rNI2HkN_VJ4qYe0cUc,1118
12
12
  upgini/version_validator.py,sha256=DvbaAvuYFoJqYt0fitpsk6Xcv-H1BYDJYHUMxaKSH_Y,1509
@@ -74,7 +74,7 @@ upgini/utils/target_utils.py,sha256=GCPn4QeJ83JJ_vyBJ3IhY5fyIRkLC9q9BE59S2FRO1I,
74
74
  upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
75
75
  upgini/utils/ts_utils.py,sha256=26vhC0pN7vLXK6R09EEkMK3Lwb9IVPH7LRdqFIQ3kPs,1383
76
76
  upgini/utils/warning_counter.py,sha256=-GRY8EUggEBKODPSuXAkHn9KnEQwAORC0mmz_tim-PM,254
77
- upgini-1.2.116.dist-info/METADATA,sha256=w7UkEj5YDpGksjUI1ii61tcOjPenO4XNEDrvPnaJVj4,50692
78
- upgini-1.2.116.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
79
- upgini-1.2.116.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
80
- upgini-1.2.116.dist-info/RECORD,,
77
+ upgini-1.2.117.dist-info/METADATA,sha256=3onXIkh96-rh_Q0DIuHUihe07upcEFukS5WsVS1R2yc,50743
78
+ upgini-1.2.117.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
79
+ upgini-1.2.117.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
80
+ upgini-1.2.117.dist-info/RECORD,,