diff --git a/prototorch/models/__init__.py b/prototorch/models/__init__.py index 2507e45..bde1ef1 100644 --- a/prototorch/models/__init__.py +++ b/prototorch/models/__init__.py @@ -1,5 +1,10 @@ from importlib.metadata import PackageNotFoundError, version +from .cbc import CBC +from .glvq import GLVQ, GMLVQ, GRLVQ, LVQMLN, ImageGLVQ, SiameseGLVQ +from .neural_gas import NeuralGas +from .vis import * + VERSION_FALLBACK = "uninstalled_version" try: __version__ = version(__name__.replace(".", "-"))