This repository contains code for a Convolutional Neural Network (CNN) model trained to classify images as happy or sad.
To use this model, follow these steps:
- Clone the repository: git clone https://github.com/Akshaj31/Happy-Sad-Model.git
- Install the required dependencies: pip install -r requirements.txt
- Run the model: python predict.py <path_to_image>
The model architecture is defined in Image Classification.ipynb. It consists of three convolutional layers followed by max-pooling layers, a flatten layer, and two dense layers.
The dataset used for training the model can be found at data. It contains images labeled as happy or sad.
To train the model, run the jupyter notebook. You can customize the training parameters in the script.
Evaluation metrics and results can also be found in the same notebook.
This project is licensed under the MIT License.