Dealing with Unbalanced Dataset for an Image Classification task


  • Implemented convolutional neural networks (CNNs) from scratch [python], and by using PyTorch on CIFAR-10 dataset

  • Trained a ResNet-32 model on unbalanced version of CIFAR-10, and handled the unbalance data using class-balanced focal loss