diff --git a/utils/Quality_Relabel.py b/utils/Quality_Relabel.py index 0b9ee52..331d34b 100644 --- a/utils/Quality_Relabel.py +++ b/utils/Quality_Relabel.py @@ -410,6 +410,12 @@ class Quality_Relabel: self.signal_select[channel][SP * self.frequency:EP * self.frequency], label=channel, color=self.color_cycle[0]) + plt_.arrow((EP + SP) // 2 * self.frequency, + self.signal_select[channel][SP * self.frequency:EP * self.frequency].max(), + (EP + SP) // 2 * self.frequency, + self.signal_select[channel][SP * self.frequency:EP * self.frequency].mean(), + ) + for j in event_code: if channel == "SpO2": mask = self.spo2_event_label[SP * self.frequency:EP * self.frequency] == j