Code for "ENHANCING CONTEXTUAL ENCODING WITH STAGE-CONFUSION AND STAGE-TRANSITION ESTIMATION FOR EEG-BASED SLEEP STAGING".

Overall architecture of the proposed model

Hyponograms of the ground trugh (top), the proposed method without auxiliary tasks (middle), and the proposed method (bottom)