     Whether PCA was used</span>
<span class="sd">    :type  PCA_used:            bool</span>
<span class="sd">    </span>
<span class="sd">    :param data_X:              The training dataset (used to save the shape)</span>
<span class="sd">    :type  data_X:              np.array</span>
<span class="sd">    </span>
<span class="sd">    :param train_loss:          The loss on the training dataset</span>
<span class="sd">    :type  train_loss:          float</span>
<span class="sd">    </span>
<span class="