Spaces:
Sleeping
Sleeping
| import re | |
| LABEL_MAP = { | |
| (1, "Normal"): 0, (1, "Abnormal"): 1, | |
| (2, "Normal"): 2, (2, "Abnormal"): 3, | |
| (3, "Normal"): 4, (3, "Abnormal"): 5, | |
| } | |
| LABEL_NAMES = { | |
| 0: "Machine 1 - Normal", 1: "Machine 1 - Abnormal", | |
| 2: "Machine 2 - Normal", 3: "Machine 2 - Abnormal", | |
| 4: "Machine 3 - Normal", 5: "Machine 3 - Abnormal", | |
| } | |
| def get_label(machine_folder, status): | |
| match = re.search(r'\d+', machine_folder) | |
| if not match: | |
| raise ValueError(f"Cannot parse machine number from: {machine_folder}") | |
| key = (int(match.group()), status) | |
| if key not in LABEL_MAP: | |
| raise ValueError(f"Unknown combination: {key}") | |
| return LABEL_MAP[key] | |