Pytorch Expert Required for a small work

Job ID: 31634541

Budget: $10 – $30 USD

Extend/improve the code to do the following:

- Graph train/val loss and accuracy per epoch (two separate graphs). Use a legend to indicate which line is train and which is val.

- Make sure you do not cause overfitting (as indicated by your plots).

- Improve the accuracy so the error on the test dataset (as computed by sklearn, we're not talking about loss here) is <17%.

- Show a confusion matrix of the prediction errors.
Related categories: Python CUDA Pytorch