def init_probabilities(self, n):
    p_z_i = np.random.randint(2, size=(n, 1)).astype(float)
    return (p_z_i, (1 - p_z_i))
def init_probabilities(self, n):
    p_z_i = np.random.randint(2, size=(n, 1)).astype(float)
    return (p_z_i, (1 - p_z_i))