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

⚡ Adds ses owanova

Browse Source
This commit is contained in:
Daniel Svitan 2024-12-15 12:16:55 +01:00
parent d3d9ebfbe2
commit 95d0625f4e
3 changed files with 29 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
analyze_living.py
View File

@ -3,7 +3,7 @@ import numpy as np
from analyze import analyze
dataset = np.load("clean.npy")
print(f"dataset shape: {dataset.shape}, analyzing column 7 (living)")
print(f"dataset shape: {dataset.shape}, analyzing column 8 (living)")
print("\t0 - with family")
print("\t1 - with family member")
print("\t2 - alone / roomates")
@ -13,7 +13,7 @@ print("")
def analyze_living(name: str, col: np.ndarray):
occupation_col = dataset[:, 7]
occupation_col = dataset[:, 8]
analyze(name, [
col[occupation_col == 0],
col[occupation_col == 1],

4
analyze_occupation.py
View File

@ -3,7 +3,7 @@ import numpy as np
from analyze import analyze
dataset = np.load("clean.npy")
print(f"dataset shape: {dataset.shape}, analyzing column 6 (occupation)")
print(f"dataset shape: {dataset.shape}, analyzing column 7 (occupation)")
print("\t0 - work hours / week >= 10")
print("\t1 - work hours / week < 10")
print("\t2 - sport")
@ -14,7 +14,7 @@ print("")
def analyze_occupation(name: str, col: np.ndarray):
occupation_col = dataset[:, 6]
occupation_col = dataset[:, 7]
analyze(name, [
col[occupation_col == 0],
col[occupation_col == 1],

25
analyze_ses.py Normal file
View File

@ -0,0 +1,25 @@
import numpy as np
from analyze import analyze
dataset = np.load("clean.npy")
print(f"dataset shape: {dataset.shape}, analyzing column 6 (ses)")
print("\t0 - lower class")
print("\t1 - middle class")
print("\t2 - upper class")
print("")
def analyze_ses(name: str, col: np.ndarray):
sex_col = dataset[:, 6]
analyze(name, [
col[sex_col == 0],
col[sex_col == 1],
col[sex_col == 2]
])
analyze_ses("gpa", dataset[:, 2])
analyze_ses("math", dataset[:, 3])
analyze_ses("slovak", dataset[:, 4])
analyze_ses("english", dataset[:, 5])