diff --git a/pyroved/models/ssivae.py b/pyroved/models/ssivae.py index bff39e3..a650065 100644 --- a/pyroved/models/ssivae.py +++ b/pyroved/models/ssivae.py @@ -3,7 +3,7 @@ ========= Semi-supervised variational autoencoder for data -with positional (rotation+translation) disorder +with orientational, positional and scale disorders Created by Maxim Ziatdinov (email: ziatdinovmax@gmail.com) """ @@ -25,7 +25,9 @@ class ssiVAE(baseVAE): """ Semi-supervised variational autoencoder with the enforcement - of rotational, translational, and scale invariances + of rotational, translational, and scale invariances. It allows performing + a classification of image/spectral data given a small number of examples + even in the presence of a distribution shift between the labeled and unlabeled parts. Args: data_dim: