Merge branch 'dev' into main

2022-01-11 18:29:32 +01:00
parent aaa3c51e0a 1a17193b35
commit 1a0e697b27
27 changed files with 537 additions and 79 deletions
--- a/prototorch/models/init.py
+++ b/prototorch/models/init.py
@@ -8,17 +8,34 @@ from .glvq import (
    GLVQ21,
    GMLVQ,
    GRLVQ,
+    GTLVQ,
    LGMLVQ,
    LVQMLN,
    ImageGLVQ,
    ImageGMLVQ,
+    ImageGTLVQ,
    SiameseGLVQ,
    SiameseGMLVQ,
+    SiameseGTLVQ,
 )
 from .knn import KNN
-from .lvq import LVQ1, LVQ21, MedianLVQ
-from .probabilistic import CELVQ, PLVQ, RSLVQ, SLVQ
-from .unsupervised import GrowingNeuralGas, HeskesSOM, KohonenSOM, NeuralGas
+from .lvq import (
+    LVQ1,
+    LVQ21,
+    MedianLVQ,
+)
+from .probabilistic import (
+    CELVQ,
+    PLVQ,
+    RSLVQ,
+    SLVQ,
+)
+from .unsupervised import (
+    GrowingNeuralGas,
+    HeskesSOM,
+    KohonenSOM,
+    NeuralGas,
+)
 from .vis import *

 __version__ = "0.4.0"