Hyper param search

[arianhosseini]

• [ ] try different z_prior thresholds
• [ ] divide w_prior term by num of dimension (i.e. 2 for spiral)