In this model, we try to predict an image whether it is number 3 or 7 from publicly avaialable dataset MNIST classified as training and validation.
This model uses cross-entropy loss function, as it classifies the number whether the given image is 3 or 7.
We are using fastai module to build our classification model using neural network.
This technique transforms the picture into 28*28 pixel size.
The loss function used in this model is cross-entropy function.
The neural network is built using PyTorch library