SaqibAMA / HOT-DOG-or-NOT-HOT-DOG

Silicon Valley inspired binary classifier to identify hot-dogs and not-hot-dogs. Source: https://www.youtube.com/watch?v=vIci3C4JkL0

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Hot Dog or Not Hot Dog

image

Silicon Valley inspired binary classifier to identify hot-dogs and not-hot-dogs. Can be used to identify anything from missiles in the air to the ones in your pants. Must be used responsibly.

Results

Version 1:

After generic training and absolutely zero effort in fine tuning, the model achieved a training accuracy of 100%, validation accuracy of 73%, and testing accuracy of 57%. Please contribute for betterment.

Version 2:

After using transfer learning with a pre-trained model - ResNet50, the accuracy of the model is increased up to 94% for the training data and 91% for the validation data.

Pull requests welcome.

Dataset Link

https://www.kaggle.com/datasets/dansbecker/hot-dog-not-hot-dog

About

Silicon Valley inspired binary classifier to identify hot-dogs and not-hot-dogs. Source: https://www.youtube.com/watch?v=vIci3C4JkL0


Languages

Language:Jupyter Notebook 100.0%