[BUGFIX] Growing neural gas

2021-06-03 15:13:27 +02:00
parent 7379c61966
commit bda88149d4
2 changed files with 5 additions and 4 deletions
--- a/prototorch/models/init.py
+++ b/prototorch/models/init.py
@@ -8,7 +8,7 @@ from .glvq import (GLVQ, GLVQ1, GLVQ21, GMLVQ, GRLVQ, LGMLVQ, LVQMLN,
                   ImageGLVQ, ImageGMLVQ, SiameseGLVQ, SiameseGMLVQ)
 from .lvq import LVQ1, LVQ21, MedianLVQ
 from .probabilistic import CELVQ, RSLVQ, LikelihoodRatioLVQ
-from .unsupervised import KNN, NeuralGas
+from .unsupervised import KNN, GrowingNeuralGas, NeuralGas
 from .vis import *

 __version__ = "0.1.7"