Merge branch 'release/0.1.4'

myautoml · Aug 10, 2020 · 3a234e6 · 3a234e6
2 parents eb2be95 + c3f2f9e
commit 3a234e6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/src/myautoml/__about__.py b/src/myautoml/__about__.py
@@ -6,7 +6,7 @@
 __summary__ = "myautoml is a package which provides a framework to automate machine learning"
 __uri__ = "https://github.com/myautoml/myautoml"
 
-__version__ = "0.1.3"
+__version__ = "0.1.4"
 
 __author__ = "Erik Jan de Vries"
 __email__ = "erikjandevries@users.noreply.github.com"

diff --git a/src/myautoml/evaluation/shap.py b/src/myautoml/evaluation/shap.py
@@ -27,13 +27,10 @@ def shap_analyse(model, x, temp_dir):
 
     shap_summary_path = save_shap_summary(temp_dir, shap_values, shap_data, shap_feature_names)
     shap_summary_bar_path = save_shap_summary_bar(temp_dir, shap_values, shap_data, shap_feature_names)
+    shap_dependence_paths = save_shap_dependence_plots(temp_dir, shap_values, shap_data, shap_feature_names, x,
+                                                       model.steps[0][1])
 
-    # TODO: Check what happens with dependence plots. Doesn't work as intended.
-    # shap_dependence_paths = save_shap_dependence_plots(temp_dir, shap_values, shap_data, shap_feature_names, x,
-    #                                                    model.steps[0][1])
-
-    paths = [shap_summary_path, shap_summary_bar_path]
-    # paths = [shap_summary_path, shap_summary_bar_path, *shap_dependence_paths]
+    paths = [shap_summary_path, shap_summary_bar_path, *shap_dependence_paths]
     artifacts = {path: 'shap' for path in paths}
 
     return tags, artifacts