n class="p">,</span>
           <span class="n">test_loss</span><span class="p">,</span>
           <span class="n">hidden_dim</span><span class="p">,</span>
           <span class="n">mse</span><span class="p">,</span>
           <span class="n">mae</span><span class="p">,</span>
           <span class="n">mde</span><span class="p">,</span>
           <span class="n">train_time</span><span class="p">]</span>

    <span class="n">ind</span> <span class="o">=</span> <span class="p">[</span><span class="nb">s