Update ssivae.py

ziatdinovmax · May 27, 2021 · 49d731d · 49d731d
1 parent cc47d0d
commit 49d731d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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: [email protected])
 """
@@ -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: