0
$\begingroup$

Here I am Showing Two Loss graphs of an Artificial Neural Network.

Model 1

enter image description here

Model 2

enter image description here

Blue -training loss

Red -val training loss

Can you help me to analyse these graphs? I read some articles and post but doesn't give me any sense.

$\endgroup$
  • 1
    $\begingroup$ The second graph definitely looks better..I suggest you go over to Andrew ngs course "Hyperparameter optimisation", you'll get it on Coursera or YouTube. $\endgroup$ – DuttaA Jun 21 at 10:34
1
$\begingroup$

Simply model 1 is a better fit compared to model 2.

  • Graph for model 1

    We notice that the training loss and validation loss aren't correlated. This means the as the training loss is decreasing, the validation loss remains the same of increases over the iterations. This means that the model is not exactly improving, but is instead overfitting the training data. This isn't what we are looking for.

  • Graph for model 2

    In this case, there is clearly a health correlation between training loss and the validation loss. They both seem to reduce and stay at a constant value. This means that the model is well trained and is equally good on the training data as well as the hidden data.

You should stick with model 2. In case you're going ahead with model 1, make sure to use the chechpoint where both the losses are at a similar value (at around 100 -150 epochs)

$\endgroup$
  • $\begingroup$ In graph 2, is there any chance of overfitting? $\endgroup$ – Majo_Jose Jun 24 at 5:26

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.