prototorch_models/prototorch/models/__init__.py

40 lines
628 B
Python
Raw Normal View History

"""`models` plugin for the `prototorch` package."""
from .callbacks import PrototypeConvergence, PruneLoserPrototypes
2021-05-21 11:11:59 +00:00
from .cbc import CBC, ImageCBC
from .glvq import (
GLVQ,
GLVQ1,
GLVQ21,
GMLVQ,
GRLVQ,
2021-11-10 17:04:24 +00:00
GTLVQ,
LGMLVQ,
LVQMLN,
ImageGLVQ,
ImageGMLVQ,
ImageGTLVQ,
SiameseGLVQ,
SiameseGMLVQ,
SiameseGTLVQ,
)
2021-06-04 20:20:32 +00:00
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 *
2022-06-01 12:25:44 +00:00
__version__ = "0.5.2"