This repository utilizes PyTorch framework in order to classify food images in a in-class Kaggle challange. Repository is a proposed solution as part of an Applied Machine Learning course at University of Amsterdam MS Information Studies: Data Science.
- Dataset_EDA.ipynb - Visualizes provided dataset in order ot inspect for any data imbalance problems.
- Dataset_Modifications.ipynb - Implements data augmentation techniques to balance class numbers.
- Ensemble_PyTorch_Model.ipynb - Implementation of simple commity of networks model ensembling.
- Kaggle_Test_Predictions.ipynb - Script which performs class prediction and outputs in nesessary Kaggle format.
- Lasagna_GAN.py - Initial implementation to battle class imbalance using GANs.
- Loss_Analysis.ipynb - Visualization of saved history objects in order to analyze loss vs accuracy dueing training epohs.