PyHealth icon indicating copy to clipboard operation
PyHealth copied to clipboard

[BMS] StageNet with Attention

Open Logiquo opened this issue 1 month ago • 0 comments

Contributor: Yongda Fan ([email protected])

Contribution Type: Model

Description Create a new model StageNet with Attention, with MHA layers inserted between SA-LSTM and SA-CNN.

Files to Review pyhealth/models/stagenet_mha.py tests/core/test_stagenet_mha.py

Logiquo avatar Dec 08 '25 01:12 Logiquo