{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "# Mixture models and label switching with MCMC {#sec-mixture-models-label-switching}\n", "\n", "[Data set download](https://s3.amazonaws.com/bebi103.caltech.edu/data/singer_transcript_counts.csv)\n", "\n", "
\\n\"+\n", " \"BokehJS does not appear to have successfully loaded. If loading BokehJS from CDN, this \\n\"+\n", " \"may be due to a slow or bad network connection. Possible fixes:\\n\"+\n", " \"
\\n\"+\n", " \"\\n\"+\n",
" \"from bokeh.resources import INLINE\\n\"+\n",
" \"output_notebook(resources=INLINE)\\n\"+\n",
" \"\\n\"+\n",
" \"\\n\"+\n \"BokehJS does not appear to have successfully loaded. If loading BokehJS from CDN, this \\n\"+\n \"may be due to a slow or bad network connection. Possible fixes:\\n\"+\n \"
\\n\"+\n \"\\n\"+\n \"from bokeh.resources import INLINE\\n\"+\n \"output_notebook(resources=INLINE)\\n\"+\n \"\\n\"+\n \"<xarray.Dataset> Size: 360kB\n",
"Dimensions: (chain: 4, draw: 1000, log10_alpha_dim_0: 2,\n",
" log10_b_dim_0: 2, alpha_dim_0: 2, b_dim_0: 2,\n",
" beta__dim_0: 2)\n",
"Coordinates:\n",
" * chain (chain) int64 32B 0 1 2 3\n",
" * draw (draw) int64 8kB 0 1 2 3 4 5 ... 994 995 996 997 998 999\n",
" * log10_alpha_dim_0 (log10_alpha_dim_0) int64 16B 0 1\n",
" * log10_b_dim_0 (log10_b_dim_0) int64 16B 0 1\n",
" * alpha_dim_0 (alpha_dim_0) int64 16B 0 1\n",
" * b_dim_0 (b_dim_0) int64 16B 0 1\n",
" * beta__dim_0 (beta__dim_0) int64 16B 0 1\n",
"Data variables:\n",
" log10_alpha (chain, draw, log10_alpha_dim_0) float64 64kB 0.3836 ....\n",
" log10_b (chain, draw, log10_b_dim_0) float64 64kB 0.8141 ... 0...\n",
" w (chain, draw) float64 32kB 0.2084 0.1267 ... 0.846 0.851\n",
" alpha (chain, draw, alpha_dim_0) float64 64kB 2.419 ... 3.082\n",
" b (chain, draw, b_dim_0) float64 64kB 6.517 29.36 ... 4.394\n",
" beta_ (chain, draw, beta__dim_0) float64 64kB 0.1534 ... 0.2276\n",
"Attributes:\n",
" created_at: 2026-01-04T07:24:37.205931+00:00\n",
" arviz_version: 0.23.0\n",
" inference_library: cmdstanpy\n",
" inference_library_version: 1.3.0<xarray.Dataset> Size: 9MB\n",
"Dimensions: (chain: 4, draw: 1000, n_ppc_dim_0: 279)\n",
"Coordinates:\n",
" * chain (chain) int64 32B 0 1 2 3\n",
" * draw (draw) int64 8kB 0 1 2 3 4 5 6 ... 993 994 995 996 997 998 999\n",
" * n_ppc_dim_0 (n_ppc_dim_0) int64 2kB 0 1 2 3 4 5 ... 273 274 275 276 277 278\n",
"Data variables:\n",
" n_ppc (chain, draw, n_ppc_dim_0) float64 9MB 167.0 20.0 ... 141.0\n",
"Attributes:\n",
" created_at: 2026-01-04T07:24:37.210195+00:00\n",
" arviz_version: 0.23.0\n",
" inference_library: cmdstanpy\n",
" inference_library_version: 1.3.0<xarray.Dataset> Size: 204kB\n",
"Dimensions: (chain: 4, draw: 1000)\n",
"Coordinates:\n",
" * chain (chain) int64 32B 0 1 2 3\n",
" * draw (draw) int64 8kB 0 1 2 3 4 5 6 ... 994 995 996 997 998 999\n",
"Data variables:\n",
" lp (chain, draw) float64 32kB -1.597e+03 ... -1.596e+03\n",
" acceptance_rate (chain, draw) float64 32kB 0.9272 0.9447 ... 0.8447 0.9545\n",
" step_size (chain, draw) float64 32kB 0.09361 0.09361 ... 0.1234\n",
" tree_depth (chain, draw) int64 32kB 3 5 5 3 5 2 3 4 ... 6 5 6 6 6 5 5\n",
" n_steps (chain, draw) int64 32kB 15 31 31 15 31 ... 71 63 63 63 31\n",
" diverging (chain, draw) bool 4kB False False False ... False False\n",
" energy (chain, draw) float64 32kB 1.598e+03 ... 1.598e+03\n",
"Attributes:\n",
" created_at: 2026-01-04T07:24:37.208542+00:00\n",
" arviz_version: 0.23.0\n",
" inference_library: cmdstanpy\n",
" inference_library_version: 1.3.0<xarray.Dataset> Size: 184B\n",
"Dimensions: (log10_alpha_dim_0: 2, log10_b_dim_0: 2, alpha_dim_0: 2,\n",
" b_dim_0: 2, beta__dim_0: 2)\n",
"Coordinates:\n",
" * log10_alpha_dim_0 (log10_alpha_dim_0) int64 16B 0 1\n",
" * log10_b_dim_0 (log10_b_dim_0) int64 16B 0 1\n",
" * alpha_dim_0 (alpha_dim_0) int64 16B 0 1\n",
" * b_dim_0 (b_dim_0) int64 16B 0 1\n",
" * beta__dim_0 (beta__dim_0) int64 16B 0 1\n",
" chain int64 8B 2\n",
" draw int64 8B 478\n",
"Data variables:\n",
" log10_alpha (log10_alpha_dim_0) float64 16B 0.3036 0.7158\n",
" log10_b (log10_b_dim_0) float64 16B 0.9057 1.517\n",
" w float64 8B 0.1839\n",
" alpha (alpha_dim_0) float64 16B 2.012 5.197\n",
" b (b_dim_0) float64 16B 8.048 32.85\n",
" beta_ (beta__dim_0) float64 16B 0.1242 0.03044\n",
"Attributes:\n",
" created_at: 2026-01-04T07:24:37.205931+00:00\n",
" arviz_version: 0.23.0\n",
" inference_library: cmdstanpy\n",
" inference_library_version: 1.3.0<xarray.DataArray 'alpha' ()> Size: 8B\n",
"array(5.1970448)\n",
"Coordinates:\n",
" chain int64 8B 2\n",
" draw int64 8B 478\n",
" alpha_dim_0 int64 8B 1| chain | draw | log10_alpha_dim_0 | log10_b_dim_0 | alpha_dim_0 | b_dim_0 | beta__dim_0 | log10_alpha | log10_b | w | alpha | b | beta_ |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| i64 | i64 | i64 | i64 | i64 | i64 | i64 | f64 | f64 | f64 | f64 | f64 | f64 |
| 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0.383558 | 0.814061 | 0.208442 | 2.4185681 | 6.517205 | 0.15344 |
| 0 | 0 | 0 | 0 | 0 | 0 | 1 | 0.383558 | 0.814061 | 0.208442 | 2.4185681 | 6.517205 | 0.034059 |
| 0 | 0 | 0 | 0 | 0 | 1 | 0 | 0.383558 | 0.814061 | 0.208442 | 2.4185681 | 29.36054 | 0.15344 |
| 0 | 0 | 0 | 0 | 0 | 1 | 1 | 0.383558 | 0.814061 | 0.208442 | 2.4185681 | 29.36054 | 0.034059 |
| 0 | 0 | 0 | 0 | 1 | 0 | 0 | 0.383558 | 0.814061 | 0.208442 | 5.756386 | 6.517205 | 0.15344 |
| log10_alpha[0] | log10_alpha[1] | log10_b[0] | log10_b[1] | w | alpha[0] | alpha[1] | b[0] | b[1] | beta_[0] | beta_[1] | chain__ | draw__ | diverging__ |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| f64 | f64 | f64 | f64 | f64 | f64 | f64 | f64 | f64 | f64 | f64 | i64 | i64 | bool |
| 0.383558 | 0.76015 | 0.814061 | 1.467764 | 0.208442 | 2.4185681 | 5.756386 | 6.517205 | 29.36054 | 0.15344 | 0.034059 | 0 | 0 | false |
| 0.494572 | 0.728771 | 0.702253 | 1.4731786 | 0.126747 | 3.1229968 | 5.3551389 | 5.0379379 | 29.728885 | 0.198494 | 0.033637 | 0 | 1 | false |
| 0.620479 | 0.689009 | 0.563716 | 1.5213664 | 0.143498 | 4.1732938 | 4.8866221 | 3.6619824 | 33.217455 | 0.273076 | 0.030105 | 0 | 2 | false |
| 0.587488 | 0.68324 | 0.562842 | 1.5254296 | 0.14857 | 3.8680115 | 4.8221473 | 3.6546186 | 33.529698 | 0.273626 | 0.029824 | 0 | 3 | false |
| 0.605264 | 0.725903 | 0.542074 | 1.491688 | 0.145304 | 4.0296225 | 5.3198928 | 3.4839687 | 31.023302 | 0.287029 | 0.032234 | 0 | 4 | false |
| log10_alpha[0] | log10_alpha[1] | log10_b[0] | log10_b[1] | w | alpha[0] | alpha[1] | b[0] | b[1] | beta_[0] | beta_[1] |
|---|---|---|---|---|---|---|---|---|---|---|
| f64 | f64 | f64 | f64 | f64 | f64 | f64 | f64 | f64 | f64 | f64 |
| 0.714835 | 0.472351 | 1.497446 | 0.716573 | 0.831534 | 5.237345 | 3.217475 | 31.716235 | 6.050639 | 0.032092 | 0.223146 |