Gaussian Mixed Model Python