Image Classification using Convolutional Neural Network (PYTORCH)
Here, I improved the prediction accuracy of an existing machine learning model created by FAHAD MEHFOOZ . I was able to archive a 7.05% improvement in the model's accuracy by tuning few hyper parameters, introducing a validation dataset to guide against overfitting, etc.