Streamer272/soc-2024
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

⚡ Adds commute owanova

Browse Source
This commit is contained in:
Daniel Svitan 2024-12-15 12:18:33 +01:00
parent 95d0625f4e
commit 7ffbde6c2c
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
analyze_commute.py Normal file
View File

@ -0,0 +1,29 @@
import numpy as np
from analyze import analyze
dataset = np.load("clean.npy")
print(f"dataset shape: {dataset.shape}, analyzing column 9 (commute)")
print("\t0 - dorms")
print("\t1 - <= 15m")
print("\t2 - <= 30m")
print("\t3 - <= 1h")
print("\t4 - > 1h")
print("")
def analyze_living(name: str, col: np.ndarray):
occupation_col = dataset[:, 9]
analyze(name, [
col[occupation_col == 0],
col[occupation_col == 1],
col[occupation_col == 2],
col[occupation_col == 3],
col[occupation_col == 4]
])
analyze_living("gpa", dataset[:, 2])
analyze_living("math", dataset[:, 3])
analyze_living("slovak", dataset[:, 4])
analyze_living("english", dataset[:, 5])