erroranalysis 0.5.3__tar.gz → 0.5.4__tar.gz

erroranalysis-0.5.4/MANIFEST.in

@@ -0,0 +1,2 @@
+include requirements.txt
+include requirements-object-detection.txt

{erroranalysis-0.5.3/erroranalysis.egg-info → erroranalysis-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: erroranalysis
-Version: 0.5.3
+Version: 0.5.4
 Summary: Core error analysis APIs
 Home-page: https://github.com/microsoft/responsible-ai-widgets
 Author: Roman Lutz, Ilya Matiach, Ke Xu
@@ -21,7 +21,7 @@ Requires-Dist: pandas<2.0.0,>=0.25.1
 Requires-Dist: scipy>=1.4.1
 Requires-Dist: scikit-learn>=0.22.1
 Requires-Dist: lightgbm>=2.0.11
-Requires-Dist: raiutils>=0.4.0
+Requires-Dist: raiutils>=0.4.2
 Provides-Extra: object-detection
 Requires-Dist: vision_explanation_methods; extra == "object-detection"
 

{erroranalysis-0.5.3 → erroranalysis-0.5.4}/erroranalysis/error_correlation_methods/gbm.py

@@ -12,8 +12,8 @@ from erroranalysis._internal.constants import ModelTask
 
 def compute_gbm_global_importance(input_data, diff, model_task,
                                   categorical_indexes):
-    """Compute global importance score for EBM between the features and error.
-    :param input_data: The input data to compute the EBM global importance
+    """Compute global importance score for GBM between the features and error.
+    :param input_data: The input data to compute the GBM global importance
         score on.
     :type input_data: numpy.ndarray
     :param diff: The difference between the label and prediction
@@ -21,7 +21,7 @@ def compute_gbm_global_importance(input_data, diff, model_task,
     :type diff: numpy.ndarray
     :param model_task: The model task.
     :type model_task: str
-    :return: The computed EBM global importance score between the features and
+    :return: The computed GBM global importance score between the features and
         error.
     :rtype: list[float]
     """
@@ -33,6 +33,11 @@ def compute_gbm_global_importance(input_data, diff, model_task,
     model.fit(input_data, diff, categorical_feature=categorical_indexes)
     explainer = shap.TreeExplainer(model)
     shap_values = explainer.shap_values(input_data)
+    dims = np.shape(shap_values)
+    # fix some inconsistencies in the shape of the shap_values
+    # for newer versions of shap>=0.45.0 for single-valued target column
+    if is_classification and len(dims) == 2:
+        shap_values = np.expand_dims(shap_values, axis=0)
     shap_mean_abs = np.abs(shap_values).mean(axis=0)
     if is_classification:
         shap_mean_abs = shap_mean_abs.mean(axis=0)

{erroranalysis-0.5.3 → erroranalysis-0.5.4}/erroranalysis/version.py

@@ -4,5 +4,5 @@
 name = 'erroranalysis'
 _major = '0'
 _minor = '5'
-_patch = '3'
+_patch = '4'
 version = '{}.{}.{}'.format(_major, _minor, _patch)

{erroranalysis-0.5.3 → erroranalysis-0.5.4/erroranalysis.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: erroranalysis
-Version: 0.5.3
+Version: 0.5.4
 Summary: Core error analysis APIs
 Home-page: https://github.com/microsoft/responsible-ai-widgets
 Author: Roman Lutz, Ilya Matiach, Ke Xu
@@ -21,7 +21,7 @@ Requires-Dist: pandas<2.0.0,>=0.25.1
 Requires-Dist: scipy>=1.4.1
 Requires-Dist: scikit-learn>=0.22.1
 Requires-Dist: lightgbm>=2.0.11
-Requires-Dist: raiutils>=0.4.0
+Requires-Dist: raiutils>=0.4.2
 Provides-Extra: object-detection
 Requires-Dist: vision_explanation_methods; extra == "object-detection"
 

{erroranalysis-0.5.3 → erroranalysis-0.5.4}/erroranalysis.egg-info/SOURCES.txt

@@ -1,6 +1,7 @@
 LICENSE
 MANIFEST.in
 README.md
+requirements-object-detection.txt
 requirements.txt
 setup.py
 erroranalysis/__init__.py

{erroranalysis-0.5.3 → erroranalysis-0.5.4}/erroranalysis.egg-info/requires.txt

@@ -3,7 +3,7 @@ pandas<2.0.0,>=0.25.1
 scipy>=1.4.1
 scikit-learn>=0.22.1
 lightgbm>=2.0.11
-raiutils>=0.4.0
+raiutils>=0.4.2
 
 [object_detection]
 vision_explanation_methods

erroranalysis-0.5.4/requirements-object-detection.txt

@@ -0,0 +1 @@
+vision_explanation_methods

{erroranalysis-0.5.3 → erroranalysis-0.5.4}/requirements.txt

@@ -3,4 +3,4 @@ pandas>=0.25.1,<2.0.0
 scipy>=1.4.1
 scikit-learn>=0.22.1
 lightgbm>=2.0.11
-raiutils>=0.4.0
+raiutils>=0.4.2

erroranalysis-0.5.3/MANIFEST.in

@@ -1 +0,0 @@
-include requirements.txt