This repository contains code for an image classification task using the Fashion MNIST dataset with TensorFlow and Keras. The project loads the dataset, normalizes the images, creates a neural network model with three layers, compiles it, and trains it on the training images and labels. The model's performance is evaluated on the test images and labels. Finally, the code predicts and visualizes the classifications for a few test images.
Image classification is a fundamental task in machine learning and computer vision. This project demonstrates how to perform clothing classification using the Fashion MNIST dataset, a popular benchmark dataset in the machine learning community. The dataset consists of 70,000 grayscale images of clothing items in 10 categories.
To run the code in this repository, you'll need to have Python installed on your system along with the following libraries:
- TensorFlow
- Keras
- NumPy
- Matplotlib
You can install these dependencies using pip:
pip install tensorflow keras numpy matplotlib