upgini 1.2.18__py3-none-any.whl → 1.2.19a1__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.18"
+__version__ = "1.2.19a1"

upgini/metrics.py

@@ -343,11 +343,16 @@ class EstimatorWrapper:
             }
             if len(average_shap_values) == 0:
                 average_shap_values = None
+            else:
+                average_shap_values = self.process_shap_values(average_shap_values)
         else:
             average_shap_values = None
 
         return self.post_process_metric(metric), average_shap_values
 
+    def process_shap_values(self, shap_values: Dict[str, float]) -> Dict[str, float]:
+        return shap_values
+
     def check_fold_metrics(self, metrics_by_fold: List[float]):
         first_metric_sign = 1 if metrics_by_fold[0] >= 0 else -1
         for metric in metrics_by_fold[1:]:
@@ -558,6 +563,13 @@ class CatBoostWrapper(EstimatorWrapper):
 
         return df, [emb_name]
 
+    def process_shap_values(self, shap_values: Dict[str, float]) -> Dict[str, float]:
+        if "__grouped_embeddings" in shap_values:
+            for emb_feature in self.emb_features:
+                shap_values[emb_feature] = shap_values["__grouped_embeddings"]
+            del shap_values["__grouped_embeddings"]
+        return shap_values
+
     def _prepare_to_calculate(self, x: pd.DataFrame, y: pd.Series) -> Tuple[pd.DataFrame, np.ndarray, dict]:
         if self.exclude_features:
             x = x.drop(columns=self.exclude_features)

{upgini-1.2.18.dist-info → upgini-1.2.19a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.18
+Version: 1.2.19a1
 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.18.dist-info → upgini-1.2.19a1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-upgini/__about__.py,sha256=XcELISZQTMy84hZtR4cdOkuOGviIP82q_-L9IH-Qu7A,23
+upgini/__about__.py,sha256=d339vH_URVWa-g9u0TOnlsaQJuX765uKQN8lSe4hfpM,25
 upgini/__init__.py,sha256=M64LwQTBa-5Jz24Zm2h8rWwlKQQ1J8nP7gGgIciS0WU,589
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=iPFiMJtk4HF1ytw9wCQr8H9RfoOKj_TIo8XYZKWgcMc,31331
@@ -7,7 +7,7 @@ upgini/features_enricher.py,sha256=3Jx6eoGULag64lN8pnwloI-RKwyLlVONrCADxpehwNo,1
 upgini/http.py,sha256=21asexflvavydzCOONJDGQBtQanCElrbnqLXakJ9Cu8,42880
 upgini/lazy_import.py,sha256=74gQ8JuA48BGRLxAo7lNHNKY2D2emMxrUxKGdxVGhuY,1012
 upgini/metadata.py,sha256=osmzdNESeh7yP3BZday6N9Q3eaIHfzhhRM1d6NSgcf0,11223
-upgini/metrics.py,sha256=0Fy_C2UAedOxyfkSe9DmZsTcsOEzSnes7QChYHZ6Euw,33913
+upgini/metrics.py,sha256=M508zOvqg0uc2sSgS8fpU7uNjGv1JA6iW_gDDOq-6PI,34474
 upgini/search_task.py,sha256=qxUxAD-bed-FpZYmTB_4orW7YJsW_O6a1TcgnZIRFr4,17307
 upgini/spinner.py,sha256=4iMd-eIe_BnkqFEMIliULTbj6rNI2HkN_VJ4qYe0cUc,1118
 upgini/version_validator.py,sha256=ddSKUK_-eGJB3NgrqOMoWJU-OxQ253WsNLp8aqJkaIM,1389
@@ -57,7 +57,7 @@ upgini/utils/sklearn_ext.py,sha256=13jQS_k7v0aUtudXV6nGUEWjttPQzAW9AFYL5wgEz9k,4
 upgini/utils/target_utils.py,sha256=qHzZRmICFbLNCrmVqGkaBcjm91L2ERRZMppci36acV4,10085
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/warning_counter.py,sha256=dIWBB4dI5XRRJZudvIlqlIYKEiwLLPcXarsZuYRt338,227
-upgini-1.2.18.dist-info/METADATA,sha256=dOuLBBNHAw4jZof-az8hEPMCT1gwPjpf_SY-rwq6_FU,48578
-upgini-1.2.18.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-upgini-1.2.18.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.18.dist-info/RECORD,,
+upgini-1.2.19a1.dist-info/METADATA,sha256=saak1tLriv8l02IV7TdY4pntedUc9Sd8qNjatTo8DNE,48580
+upgini-1.2.19a1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.2.19a1.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.19a1.dist-info/RECORD,,