Skip to content

Commit 358b966

Browse files
committed
Add docs
1 parent b6ba3be commit 358b966

File tree

2 files changed

+26
-0
lines changed

2 files changed

+26
-0
lines changed

docs/api/interpret.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ Available Methods
4141
.. toctree::
4242
:maxdepth: 3
4343

44+
interpret/pyhealth.interpret.methods.gim
4445
interpret/pyhealth.interpret.methods.chefer
4546
interpret/pyhealth.interpret.methods.deeplift
4647
interpret/pyhealth.interpret.methods.integrated_gradients
Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
pyhealth.interpret.methods.gim
2+
================================
3+
4+
Overview
5+
--------
6+
7+
The Gradient Interaction Modifications (GIM) interpreter adapts the StageNet
8+
attribution method described by Edin et al. (2025). It recomputes softmax
9+
gradients with a higher temperature so that token-level interactions remain
10+
visible when cumulative softmax layers are present.
11+
12+
Use this interpreter with StageNet-style models that expose
13+
``forward_from_embedding`` and ``embedding_model``.
14+
15+
For a complete working example, see:
16+
``examples/gim_stagenet_mimic4.py``
17+
18+
API Reference
19+
-------------
20+
21+
.. autoclass:: pyhealth.interpret.methods.GIM
22+
:members:
23+
:undoc-members:
24+
:show-inheritance:
25+
:member-order: bysource

0 commit comments

Comments
 (0)