t-ipython3 notranslate"><div class="highlight"><pre>
<span></span><span class="c1"># sampling parameters</span>
<span class="n">n_iter</span> <span class="o">=</span> <span class="mi">2000</span>
<span class="n">n_chains</span> <span class="o">=</span> <span class="mi">2</span>
<span class="n">n_thin</span> <span class="o">=</span> <span class="mi">1</span>

<span class="c1"># learning parameters</span>
<span class="n">K</span> <span class="o">=</span> <span class="mi">4</span> <span class="c1"># n options 