Skip to content

Using XGBoost algorithm with VGG16's pre-trained features to train an image classifier.

Notifications You must be signed in to change notification settings

kalideir/XGBoost-Image-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

XGBoost-Image-Classification

A simple 3-class classification task. Used a pretrained-model (VGG16) for initial weight extraction and then used these weight in XGBoost classifier.

About

Using XGBoost algorithm with VGG16's pre-trained features to train an image classifier.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published