</span>
<span class="sd">    :param model:           The neural network model</span>
<span class="sd">    :type  model:           nn.module</span>

<span class="sd">    :param data_X:          The training dataset</span>
<span class="sd">    :type  data_X:          np.array</span>

<span class="sd">    :param data_y:          the target dataset</span>
<span class="sd">    :type  data_y:          np.array</span>

<span class="sd">    :param n_epochs:        The number of epochs of training</span>
<span class