Iris Flower Classifier

A deep learning model using PyTorch to classify Iris flowers into Setosa, Versicolor, or Virginica.

MNIST Example
Loss After 100 Epochs

Project Overview

This project uses a neural network built with PyTorch to classify flowers from the Iris dataset based on petal and sepal dimensions.

  • Dataset preprocessing and normalization
  • Implementation of a neural network with 3 layers
  • Training and evaluation with PyTorch
  • Visualization of results using Matplotlib
View Code on GitHub

Get in Touch

Looking to collaborate or have a project in mind? Reach out to me via email or social media.

EmailLinkedInGitHub