dltk.models.autoencoder package

dltk.models.autoencoder.convolutional_autoencoder module

class dltk.models.autoencoder.convolutional_autoencoder.ConvolutionalAutoencoder(num_hidden_units=2, filters=(16, 64, 128, 256), strides=((1, 1, 1), (2, 2, 2), (1, 1, 1), (2, 2, 2)), relu_leakiness=0.01, name='conv_ae')

Bases: dltk.core.modules.base.AbstractModule

Convolutional Autoencoder

This module builds a convolutional autoencoder with varying number of layers and hidden units.

Module contents