I have implemented a stan hierarchical model with level 1 to be a linear model and level 2 Gaussian mixture model. The models has convergence problem. Any comments on improving the model.

```
data {
int<lower=0> J; # No of subjects
int<lower=0> N; # No of observations
int<lower=1,upper=J> RID[N];
vector[N] x; # Cognitive Score
vector[N] y; # Output Label - Disease Stage
}
parameters{
vector[J] a;
real mu_a;
vector[J] b;
real mu_b;
real<lower=0,upper=1> sigma_a;
real<lower=0,upper=1> sigma_b;
# Gaussian Parameters for level 2
vector[2] mu;
real<lower=0> sigma[2];
real<lower=0, upper=1> theta;
}
transformed parameters {
vector[N] y_hat;
for(i in 1:N)
y_hat[i] = a[RID[i]] + x[i] * b[RID[i]];
}
model {
sigma_a ~ uniform(0, 1);
a ~ normal(mu_a, sigma_a);
sigma_b ~ uniform(0, 1);
b ~ normal(mu_b, sigma_b);
sigma ~ normal(0, 1);
mu ~ normal(0, 1);
theta ~ beta(5, 5);
target += log_mix(theta,
normal_lpdf(y_hat | mu[1], sigma[1]),
normal_lpdf(y_hat | mu[2], sigma[2]));
}
```