AiOrNot / README.md
miniMaddy's picture
Create README.md
d057d3d
# AlexNet
Submission for coding assignment of Fatima Fellowship.
## Model Details
This is a simple implementation of AlexNet in Keras for the coding assignment of Fatima Fellowship.
### Model Description
- **Developed by:** Madhav Kumar
- **Model type:** Convolutional Neural Network
## How to Get Started with the Model
The files contains the notebook Alexnet_without_cropping.ipynb. It contains the code that was used to train the model. In the files, "weights-improvement-07-0.87.hdf5" contain the weights of the best model during training.
"logs-20230326T161429Z-001.zip" contain the logs of the training.
## Training Details
### Training Data
The files contain the training data images in train.zip and the data with image names and labels in the train.csv.
### Training Procedure
I trained the AlexNet model on the training data on Google Colab.
#### Preprocessing
I used data augmentaion to increase the size of training data.
## Evaluation
The model was tested on a validation data which was 20% percent of the train.zip data. The maximum accuracy obtained was 86.88%.