Return the histogram of the training
Maxime-POULAIN-Verlingue opened this issue · comments
Hello,
Description of Problem: We wanted to get the values of metrics (like accuracy) from the train function. It would give us the possibility to plot the evolution of the accuracy during the different epochs of the training, or get the last value as the accuracy on the training dataset.
Overview of the Solution: To do that, we just need to store the output of the fit function in the train function of melusine. Then we return the value of this output.
Hi Maxime,
This issue is interesting and represent low risks to integrate.
Would you be willing to submit a PR (and link it to this issue)?
Best regards,