/CNN-CIFAR-10

Building a Convolutional Neural Network on CIFAR-10 dataset

Primary LanguageJupyter Notebook

CIFAR-10 Dataset Image Classification using CNN

Dataset on kaggle

Contents

Data Preprocessing

  • Scaling
  • Encoding

Modeling

We use the following architecture for our model

m

Model Improving

Here we use some techniques to avoid overfitting and increase the accuracy

  • Early stopping
  • Data augmentation

Evaluation

We use precision, recall and accuracy as metrics in our model

  • Classification report

  • Confusion matrix

  • Plotting

  • p