armed.models.autoencoder_classifier

Autoencoder-classifiers, including domain adversarial and mixed effects variations.

Functions

load_weights_base_aec(model, weights_path)

Loads weights into BaseAutoencoderClassifier.

Classes

AdversarialClassifier(*args, **kwargs)
AuxClassifier(*args, **kwargs)
BaseAutoencoderClassifier(*args, **kwargs)
Decoder(*args, **kwargs)
DomainAdversarialAEC(*args, **kwargs)
DomainEnhancingAutoencoderClassifier(*args, ...)
Encoder(*args, **kwargs)
ImageClassifier(*args, **kwargs)
LatentClassifier(*args, **kwargs)
MixedEffectAuxClassifier(*args, **kwargs)
MixedEffectsAEC(*args, **kwargs)	Still under development!
RandomEffectDecoder(*args, **kwargs)
RandomEffectEncoder(*args, **kwargs)
TiedConv2DTranspose(*args, **kwargs)
TiedDecoder(*args, **kwargs)