Project for the University Master Degree Course of Statistical Methods for Machine Learning.
The purpose of this project is to accurately classify images of muffins and chihuahuas, utilizing different neural network architectures to solve the task.
The projects described in this document refer to the Muffin vs chihuahua dataset, which is published on Kaggle and released under the CC0 license: Public Domain.
kaggle
tensorflow
keras-tuner
pandas
numpy
matplotlib
To install the required libraries, run:
pip install kaggle tensorflow keras-tuner pandas numpy matplotlib
-
Download the Dataset:
- Ensure that your
kaggle.json
file is properly configured for the Kaggle API. - Download the dataset using the following command:
kaggle datasets download -d samuelcortinhas/muffin-vs-chihuahua-image-classification
- Unzip the dataset and place it in the appropriate directory.
- Ensure that your
-
Running the Model:
- Run the provided Jupyter Notebook to train and evaluate the models.
- The training results, including accuracy and loss curves, will be displayed.
-
Hyperparameter Tuning:
- Use Keras Tuner to optimize the model's hyperparameters. Adjust the search space in the script as needed.
Neural_Network_Project.ipynb
: Contains the Jupyter notebooks for data exploration and model training.StatisticalMethods_Report.pdf
: Contains the Report with the description of the results