Refactor non-gradient-lvq models into lvq.py

2021-05-25 20:37:34 +02:00
parent 32d6f95db0
commit d411e52be4
4 changed files with 96 additions and 89 deletions
--- a/prototorch/models/init.py
+++ b/prototorch/models/init.py
@@ -2,8 +2,9 @@ from importlib.metadata import PackageNotFoundError, version

 from . import probabilistic
 from .cbc import CBC, ImageCBC
-from .glvq import (GLVQ, GLVQ1, GLVQ21, GMLVQ, GRLVQ, LVQ1, LVQ21, LVQMLN,
-                   ImageGLVQ, ImageGMLVQ, SiameseGLVQ)
+from .glvq import (GLVQ, GLVQ1, GLVQ21, GMLVQ, GRLVQ, LVQMLN, ImageGLVQ,
+                   ImageGMLVQ, SiameseGLVQ)
+from .lvq import LVQ1, LVQ21, MedianLVQ
 from .unsupervised import KNN, NeuralGas
 from .vis import *