Skip to content

Stacked Generalisation Model for Pokemon Classification

Notifications You must be signed in to change notification settings

jaredquekjz/pokemonClassifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PokemonClassifier in Tensorflow 2.0

The program trains 4 DNNs that are in turn stack-generalised to achieve a final accuracy of ~75-85%.

The first 3 intermediate DNNs stacks Dense layers with the frozen convolutional layers (pre-trained features) from MobileNet, ResNet and VGG16. The 4th DNN is adapted from an example from Francois Chollet.

This program is developed as part of an experiment in metalearning.

Pokemon data set includes 598 images split into 10 classes adapted from: https://github.com/rileynwong/pokemon-images-dataset-by-type

About

Stacked Generalisation Model for Pokemon Classification

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages