finished notebook

basf · Dec 2, 2024 · ccb43e3 · ccb43e3
1 parent 0c14d58
commit ccb43e3
Show file tree

Hide file tree

Showing 2 changed files with 112 additions and 475 deletions.
diff --git a/molpipeline/explainability/visualization/visualization.py b/molpipeline/explainability/visualization/visualization.py
@@ -472,7 +472,7 @@ def structure_heatmap_shap(  # pylint: disable=too-many-branches
 
         # note: the prediction/expected value of the last array element is used
         text = (
-            f"$P(y=1|X) = {explanation.prediction[-1]:.2f}$ ="
+            f"$Prediction = {explanation.prediction[-1]:.2f}$ ="
             "\n"
             "\n"
             f"  $expected \ value={explanation.expected_value[-1]:.2f}$   +   "  # noqa: W605 # pylint: disable=W1401