Skip to content

Aristotelis1/AutoencoderCNN-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

67 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AutoencoderNN-Classifier

Train your autoencoder:

$python test_autoencoder.py –d <#dataset>

Default Autoencoder layer-block:

-----functions.py--------

Encoder: Convolutional layer followed by Batch Normalizaiton. After the second normalization, there is a downsampling (Maxpooling). Dropout layers have been added after the second layer-block. Decoder: Decoder part is a mirror of the encoder with a sigmoid as an activation function for the last convolutional layer.

Train your classification model:

$python test_classification.py –d <#training set> –dl <#training labels> -t <#testset> -tl <#test labels> -model <#autoencoder model>

Taking the encoder part of the autoencoder and adds the following layers: ->Flatten layer ->Dense layer ->Batch Normalization ->Dropout layer ->Dense(units = 10)

Accuracy: 0,9952

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages