diff --git a/prototorch/models/__init__.py b/prototorch/models/__init__.py index ba29fef..cd3946d 100644 --- a/prototorch/models/__init__.py +++ b/prototorch/models/__init__.py @@ -1,3 +1,39 @@ """`models` plugin for the `prototorch` package.""" +from .callbacks import PrototypeConvergence, PruneLoserPrototypes +from .cbc import CBC, ImageCBC +from .glvq import ( + GLVQ, + GLVQ1, + 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, + RSLVQ, + SLVQ, +) +from .unsupervised import ( + GrowingNeuralGas, + KohonenSOM, + NeuralGas, +) +from .vis import * + __version__ = "1.0.0-a1"