an class="o">.</span><span class="n">time</span><span class="p">()</span> <span class="o">-</span> <span class="n">start_time</span><span class="p">),</span>
                         <span class="bp">self</span><span class="o">.</span><span class="n">best_val_score</span><span class="p">))</span>

            <span class="c1"># Plotting predictions and training metrics</span>
            <span class="k">if</span> <span class="n">epoch</span> <span class="o">%</span> <span class="bp">self</span><span class="