prototorch_models/prototorch/models/__init__.py

27 lines
609 B
Python
Raw Normal View History

"""`models` plugin for the `prototorch` package."""
2021-04-23 15:27:47 +00:00
from importlib.metadata import PackageNotFoundError, version
2021-04-21 11:13:28 +00:00
from .callbacks import (EarlyStopWithoutVal, PrototypeConvergence,
PruneLoserPrototypes)
2021-05-21 11:11:59 +00:00
from .cbc import CBC, ImageCBC
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
2021-05-21 13:42:45 +00:00
from .unsupervised import KNN, NeuralGas
from .vis import *
2021-05-11 15:18:29 +00:00
__version__ = "0.1.7"