diff --git a/draw_tools/draw_label.py b/draw_tools/draw_label.py index 134202e..6b0dcb9 100644 --- a/draw_tools/draw_label.py +++ b/draw_tools/draw_label.py @@ -125,7 +125,7 @@ def plt_signal_label_on_ax(ax: Axes, signal_data, segment_start, segment_end, ev color='green', alpha=0.8, label='Amplitude Change Mask') for event_code in event_codes: sa_mask = event_mask["SA_Label"][segment_start:segment_end].repeat(signal_fs) == event_code - score_mask = event_mask["SA_Score_Alpha"][segment_start:segment_end].repeat(signal_fs) + score_mask = event_mask["SA_Score"][segment_start:segment_end].repeat(signal_fs) # y = (sa_mask * score_mask).astype(float) y = (chn_signal[segment_start * signal_fs:segment_end * signal_fs] * sa_mask).astype(float) np.place(y, y == 0, np.nan) @@ -191,8 +191,10 @@ def draw_psg_bcg_label(psg_data, psg_label, bcg_data, event_mask, segment_list, if mask.startswith("Resp_") or mask.startswith("BCG_"): event_mask[mask] = utils.none_to_nan_mask(event_mask[mask], 0) - event_mask["SA_Score_Alpha"] = score_mask2alpha(event_mask["SA_Score"]) - event_mask["SA_Score_Alpha"] = utils.none_to_nan_mask(event_mask["SA_Score_Alpha"], 0) + event_mask["SA_Score"] = utils.none_to_nan_mask(event_mask["SA_Score"], 0) + + # event_mask["SA_Score_Alpha"] = score_mask2alpha(event_mask["SA_Score"]) + # event_mask["SA_Score_Alpha"] = utils.none_to_nan_mask(event_mask["SA_Score_Alpha"], 0) fig, axes = create_psg_bcg_figure() for segment_start, segment_end in tqdm(segment_list): @@ -227,8 +229,8 @@ def draw_resp_label(resp_data, resp_label, segment_list): if mask.startswith("Resp_"): resp_label[mask] = utils.none_to_nan_mask(resp_label[mask], 0) - resp_label["Resp_Score_Alpha"] = score_mask2alpha(resp_label["Resp_Score"]) - resp_label["Resp_Label_Alpha"] = utils.none_to_nan_mask(resp_label["Resp_Label_Alpha"], 0) + # resp_label["Resp_Score_Alpha"] = score_mask2alpha(resp_label["Resp_Score"]) + # resp_label["Resp_Label_Alpha"] = utils.none_to_nan_mask(resp_label["Resp_Label_Alpha"], 0) fig, axes = create_resp_figure() for segment_start, segment_end in segment_list: