Joint Probability Matrices - Data Generation

joint_probability_matricies_1.py

# Data Generation

NUM_STUDENTS = 30
MEAN = 80
STD = 20

# https://stackoverflow.com/questions/36894191/how-to-get-a-normal-distribution-within-a-range-in-numpy
# Need to cap the values of the distributions to [0,100]
def get_truncated_normal(mean, sd, size, low, upp):
    return truncnorm((low - mean) / sd, (upp - mean) / sd, loc=mean, scale=sd).rvs(size)

grades_GT = get_truncated_normal(MEAN, STD, NUM_STUDENTS, 0, 100)
grades_P = get_truncated_normal(MEAN, STD, NUM_STUDENTS, 0, 100)