Try this gist as an example: estimate-means-missing-gaussian · GitHub. You are almost right with the model you had. It was just missing some shape info, I think.
Try this gist as an example: estimate-means-missing-gaussian · GitHub. You are almost right with the model you had. It was just missing some shape info, I think.