Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
621377f
feat: Add PTB-XL dataset and ProtoECGNet model for multi-label ECG c…
irvifa Oct 21, 2026
acded13
fix: address AI review feedback on ProtoECGNet and PTB-XL
irvifa Oct 24, 2026
9a4bc14
docs: include survival_mdn and ProtoECGNet in models API reference
irvifa Oct 24, 2026
9e28dcb
Merge branch 'master' into protoecgnet-ptbxl
irvifa Oct 24, 2026
b9a9d3a
fix: address optional review feedback on ProtoECGNet and PTB-XL
irvifa Oct 26, 2026
7bdd110
fix: address optional review feedback on ProtoECGNet and PTB-XL
irvifa Oct 26, 2026
8916bba
fix: write PTB-XL metadata CSV to cache_dir to support read-only roots
irvifa Oct 26, 2026
b6d1ee8
Merge master (PTB-XL #1187) and update task/model to master dataset API
irvifa Oct 27, 2026
f1610fd
Four blocking issues fixed:
irvifa Oct 27, 2026
ca0f9a3
chore: tighten PR per self-review before next bot cycle
irvifa Oct 27, 2026
be90e67
fix: address bot review on class_weights buffer, interpretability
irvifa Oct 27, 2026
4dd101a
fix: address second bot review on state_dict, interpretability
irvifa Oct 27, 2026
f290e48
fix: address third bot review — single decoder, superclass reuse,
irvifa Oct 27, 2026
2216e3a
fix: address fourth bot review — stable class_weights shape under
irvifa Oct 28, 2026
968f2cb
fix: persist training stage in state_dict so learned-fusion
irvifa Oct 28, 2026
fee6fd0
fix: route logits through learned fusion head in the "all" stage;
irvifa Oct 29, 2026
2d51ace
Merge remote-tracking branch 'upstream/master' into protoecgnet-ptbxl
irvifa Oct 30, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions docs/api/models.rst
Original file line number Diff line number Diff line change
Expand Up @@ -254,3 +254,5 @@ API Reference
models/pyhealth.models.califorest
models/pyhealth.models.XGBoostModel
models/pyhealth.models.survival_mdn
models/pyhealth.models.ProtoECGNet

26 changes: 26 additions & 0 deletions docs/api/models/pyhealth.models.ProtoECGNet.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
pyhealth.models.ProtoECGNet
===================================

ProtoECGNet: case-based interpretable multi-label ECG classification with a
prototype loss that includes a label co-occurrence contrastive term
(`Sethi et al., MLHC 2025 <https://proceedings.mlr.press/v298/sethi25a.html>`_).

.. autoclass:: pyhealth.models.ProtoECGNet
:members:
:undoc-members:
:show-inheritance:

.. autoclass:: pyhealth.models.PrototypeBranch
:members:
:undoc-members:
:show-inheritance:

.. autoclass:: pyhealth.models.ResNet1dBackbone
:members:
:undoc-members:
:show-inheritance:

.. autoclass:: pyhealth.models.ResNet2dBackbone
:members:
:undoc-members:
:show-inheritance:
1 change: 1 addition & 0 deletions docs/api/tasks.rst
Original file line number Diff line number Diff line change
Expand Up @@ -215,6 +215,7 @@ Available Tasks
Cardiology Detection <tasks/pyhealth.tasks.cardiology_detect>
COVID-19 CXR Classification <tasks/pyhealth.tasks.COVID19CXRClassification>
COVID-19 CXR Image Generation <tasks/pyhealth.tasks.COVID19CXRImageGeneration>
PTB-XL Multilabel Classification <tasks/pyhealth.tasks.PTBXLMultilabelClassification>
DKA Prediction (MIMIC-IV) <tasks/pyhealth.tasks.dka>
Drug Recommendation <tasks/pyhealth.tasks.drug_recommendation>
EHR Generation <tasks/pyhealth.tasks.generate_ehr>
Expand Down
10 changes: 10 additions & 0 deletions docs/api/tasks/pyhealth.tasks.PTBXLMultilabelClassification.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
pyhealth.tasks.PTBXLMultilabelClassification
============================================

Multi-label classification of PTB-XL 12-lead ECGs (all 71 SCP statements, or the
diagnostic, sub/super-diagnostic, form or rhythm label sets).

.. autoclass:: pyhealth.tasks.PTBXLMultilabelClassification
:members:
:undoc-members:
:show-inheritance:
Loading
Loading