Binary-Class Image Classification Model for Pizza vs. Ice Cream Using TensorFlow Take 3

SUMMARY: This project aims to construct a predictive model using a TensorFlow convolutional neural network (CNN) and document the end-to-end steps using a template. The Pizza vs. Ice Cream dataset is a binary classification situation where we attempt to predict one of two possible outcomes.

INTRODUCTION: The dataset contains over 1,000 images of pizza and ice cream. The researcher collected these images to investigate the machine learning model’s ability to understand and distinguish the basic features of pizza and ice cream.

ANALYSIS: The MobileNetV2 model’s performance achieved an accuracy score of 96.15% after five epochs using the validation dataset. When we applied the model to the test dataset, the model achieved an accuracy score of 96.22%.

CONCLUSION: In this iteration, the TensorFlow MobileNetV2 CNN model appeared suitable for modeling this dataset.

Dataset ML Model: Binary classification with numerical features

Dataset Used: Pizza vs. Ice Cream

Dataset Reference:

One source of potential performance benchmarks:

The HTML formatted report can be found here on GitHub.