span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">model</span><span class="p">,</span> <span class="n">data_X</span><span class="p">,</span> <span class="n">data_y</span><span class="p">,</span>
                 <span class="n">n_epochs</span><span class="p">,</span>
                 <span class="n">optimiser</span><span class="p">,</span>
                 <span class="n">batch_size</span><span class="p">,</span>
                 <span class="n">loss_function</span>