An open API service indexing awesome lists of open source software.

https://github.com/mehrab-kalantari/cnn-cifar-10

Building a Convolutional Neural Network on CIFAR-10 dataset
https://github.com/mehrab-kalantari/cnn-cifar-10

cifar-10 cnn convolutional-neural-networks data-augmentation deep-learning

Last synced: 4 months ago
JSON representation

Building a Convolutional Neural Network on CIFAR-10 dataset

Awesome Lists containing this project

README

          

# CIFAR-10 Dataset Image Classification using CNN

[Dataset on kaggle](https://www.kaggle.com/datasets/fedesoriano/cifar10-python-in-csv)

## Contents
### Data Preprocessing
* Scaling
* Encoding

### Modeling
We use the following architecture for our model

![m](sample/model.png)

### 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](sample/output.png)