Skip to content

Commit

Permalink
explainability: new module (#44)
Browse files Browse the repository at this point in the history
* explainability: new module

    - Add Explainer class and explanation
      data structures to express explanations for feature
      vectors and molecules.
    - Add Christian W. Feldmanns visualization code for shap
      weighted heatmaps of the molecular structure.
    - Add as experimental because we might change the API.
      However, the methodology is fully functional.
  • Loading branch information
JochenSiegWork authored Jan 31, 2025
1 parent 2c603aa commit 988bea1
Show file tree
Hide file tree
Showing 19 changed files with 4,143 additions and 0 deletions.
23 changes: 23 additions & 0 deletions molpipeline/experimental/explainability/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
"""Explainability module for the molpipeline package."""

from molpipeline.experimental.explainability.explainer import (
SHAPKernelExplainer,
SHAPTreeExplainer,
)
from molpipeline.experimental.explainability.explanation import (
SHAPFeatureAndAtomExplanation,
SHAPFeatureExplanation,
)
from molpipeline.experimental.explainability.visualization.visualization import (
structure_heatmap,
structure_heatmap_shap,
)

__all__ = [
"SHAPTreeExplainer",
"SHAPKernelExplainer",
"SHAPFeatureExplanation",
"SHAPFeatureAndAtomExplanation",
"structure_heatmap",
"structure_heatmap_shap",
]
Loading

0 comments on commit 988bea1

Please sign in to comment.