Exemplo n.º 1
0
def set_scorer_times(scorer: Autoscorer, nrem_start_time: int = 4940,
                          nrem_end_time: int = 5000, rem_start_time: int = 5000,
                          rem_end_time: int = 5060) -> None:
    scorer.nrem_start_time = nrem_start_time
    scorer.nrem_end_time = nrem_end_time
    scorer.rem_start_time = rem_start_time
    scorer.rem_end_time = rem_end_time
    scorer.baseline_dict = {'RSWA_P': {'REM_0': {}, 'REM_1':{}}, 'RSWA_T': {}}
Exemplo n.º 2
0
def set_scorer_baseline_dict(scorer: Autoscorer, rem_subsequences: int = 2,
                             channels: list = ['Chin', 'L Leg', 'R Leg']) -> None:
    baseline_dict = {'RSWA_P': {}, 'RSWA_T': {}}
    for i in range(rem_subsequences):
        baseline_dict['RSWA_P'][f'REM_{i}'] = {}
        baseline_dict['RSWA_T'][f'REM_{i}'] = {}
        for channel in channels:
            baseline_dict['RSWA_T'][f'REM_{i}'][channel] = {}
    scorer.baseline_dict = baseline_dict