🔨 Changes sleep enum

This commit is contained in:
Daniel Svitan 2024-12-15 12:26:58 +01:00
parent cb4241ff97
commit 7a96f1e82c
3 changed files with 7 additions and 7 deletions

View File

@ -66,7 +66,7 @@ The cleaned dataset will have the following structure:
### Sleep
```
0 - long sleepers
0 - short sleepers
1 - medium sleepers
2 - short sleepers
2 - long sleepers
```

View File

@ -4,9 +4,9 @@ from analyze import analyze
dataset = np.load("clean.npy")
print(f"dataset shape: {dataset.shape}, analyzing column 10 (sleep)")
print("\t0 - long sleepers")
print("\t0 - short sleepers")
print("\t1 - medium sleepers")
print("\t2 - short sleepers")
print("\t2 - long sleepers")
print("")

View File

@ -112,11 +112,11 @@ def parse_commute(txt: str) -> int:
def parse_sleep(txt: str) -> int:
match txt:
case "9 hodín a viac":
case "6 hodín a menej":
return 0
case "7 až 9 hodín":
return 1
case "6 hodín a menej":
case "9 hodín a viac":
return 2
case _:
print("ERROR: Coudln't determine sleep")
@ -132,7 +132,7 @@ def parse_absence(txt: str) -> float:
txt = input("Please enter fixed value: ")
for i in range(1, len(df)):
for i in range(len(arr)):
row = arr[i]
current = []