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.
-
Notifications
You must be signed in to change notification settings - Fork 0
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…
beetrandahiya/Cats-VS-Dogs
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
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 0
No packages published