class="o">.</span><span class="n">model</span><span class="p">(</span><span class="n">nx</span><span class="o">+</span><span class="n">eta</span><span class="p">)</span>
        <span class="n">py</span> <span class="o">=</span> <span class="n">out</span><span class="o">.</span><span class="n">max</span><span class="p">(</span><span class="mi">1</span><span class="p">)[</span><span class="mi">1</span><span class="p">]</span>
        <span class="n">ny</span> <span class="o">=</span> <span class="n">out</s