Skip to content

Apply AlexNet CNN architecture to predict Pokemon types.

Notifications You must be signed in to change notification settings

yylmax/MachineLearningCNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

MachineLearning

This project is about recognizing main pokemon type(water, fire, grass). Convolutional neural network model used is Alexnet. Some visulizations potions and data augmentation are done before CNN model.

The choice of number of epochs is based on the experiment output(accuracy stops increasing). Hyperparameter tuning includes changes in batch size, different model optimizer and learning rate, droupout layers, etc.

Final ouptput is considerable since limited datas are provided to the model. dataset link: https://www.kaggle.com/vishalsubbiah/pokemon-images-and-types

About

Apply AlexNet CNN architecture to predict Pokemon types.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published