stan demos

jachymb · Jul 30, 2024 · 6456b2d · 6456b2d
1 parent 16cef92
commit 6456b2d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/assets/stan_demos/imdb_inference.stan b/assets/stan_demos/imdb_inference.stan
@@ -0,0 +1,11 @@
+data {
+  vector<lower=0>[5] alpha;
+  array[5] int rating;
+}
+parameters {
+  simplex[5] d;
+}
+model {
+  d ~ dirichlet(alpha);
+  ratings[i] ~ multinomial(d);
+}