Skip to content

This refers to pytorch implementation of semi-supervised learning. The idea is to have few labeled dataset, say 10 examples per class, and a large unlabeled dataset, to improve accuracy on the test set.

Notifications You must be signed in to change notification settings

nisarnabeel/semi-supervised-GAN-for-mnist-image-classification-20--increase

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

semi-supervised GAN for mnist image classification 20% increase

This refers to pytorch implementation of semi-supervised learning. The idea is to have few labeled dataset, say 10 examples per class, and a large unlabeled dataset, to improve accuracy on the test set. For this reason, the discriminator of the GAN is turned to classifier plus regular discriminator. Notably, a custom activation function is used, refer to https://arxiv.org/abs/1606.03498. Using 10 labeled examples, 20% increase in accuracy is seen on the test set.

About

This refers to pytorch implementation of semi-supervised learning. The idea is to have few labeled dataset, say 10 examples per class, and a large unlabeled dataset, to improve accuracy on the test set.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages