Staat klaar om een grafiek te maken alleen epochs ff aanpassen aan iets
meer en de test kan je eventueel veranderen