sd">    :param val_loss:            The loss on the validation dataset</span>
<span class="sd">    :type  val_loss:            float</span>
<span class="sd">    </span>
<span class="sd">    :param test_loss:           The loss on the test dataset</span>
<span class="sd">    :type  test_loss:           float</span>
<span class="sd">    </span>
<span class="sd">    :param train_time:          The amount of time to train</span>
<span class="sd">    :type  train_time:          float</span>
<span class="sd">    