Skip to content

This is my first nice machine learning model, This model gave a 97.85% accuracy in classifying between Cats and Dogs. I made it using a pre-trained base model MobileNet V2 , and after that i added a global average pooling and then a dense layer for categorization between two classes ( cats and dogs) , i used only one dense neuron in last layer e…

Notifications You must be signed in to change notification settings

beetrandahiya/Cats-VS-Dogs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 

Repository files navigation

Cats-VS-Dogs

This is my first nice machine learning model, This model gave a 97.85% accuracy in classifying between Cats and Dogs. I made it using a pre-trained base model MobileNet V2 , and after that i added a global average pooling and then a dense layer for categorization between two classes ( cats and dogs) , i used only one dense neuron in last layer even when i had to classify 2 classes because , as we know that in test data there will only be two type of images cats and dogs , so we can just classify dogs and then ones that doesn't qualify the classification are cats. It has been trained on a small dataset to show how a machine learning model can work well even on a small dataset if we use pre-trained models inn architecture.

About

This is my first nice machine learning model, This model gave a 97.85% accuracy in classifying between Cats and Dogs. I made it using a pre-trained base model MobileNet V2 , and after that i added a global average pooling and then a dense layer for categorization between two classes ( cats and dogs) , i used only one dense neuron in last layer e…

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published