2021-06-01 21:44:16 +00:00
|
|
|
"""`models` plugin for the `prototorch` package."""
|
|
|
|
|
2021-06-03 11:38:16 +00:00
|
|
|
from .callbacks import PrototypeConvergence, PruneLoserPrototypes
|
2021-05-21 11:11:59 +00:00
|
|
|
from .cbc import CBC, ImageCBC
|
2021-06-14 18:08:08 +00:00
|
|
|
from .glvq import (
|
|
|
|
GLVQ,
|
|
|
|
GLVQ1,
|
|
|
|
GLVQ21,
|
|
|
|
GMLVQ,
|
|
|
|
GRLVQ,
|
|
|
|
LGMLVQ,
|
|
|
|
LVQMLN,
|
|
|
|
ImageGLVQ,
|
|
|
|
ImageGMLVQ,
|
|
|
|
SiameseGLVQ,
|
|
|
|
SiameseGMLVQ,
|
|
|
|
)
|
2021-06-04 20:20:32 +00:00
|
|
|
from .knn import KNN
|
2021-05-25 18:37:34 +00:00
|
|
|
from .lvq import LVQ1, LVQ21, MedianLVQ
|
2021-06-08 13:01:08 +00:00
|
|
|
from .probabilistic import CELVQ, PLVQ, RSLVQ, SLVQ
|
2021-06-07 16:44:15 +00:00
|
|
|
from .unsupervised import GrowingNeuralGas, HeskesSOM, KohonenSOM, NeuralGas
|
2021-05-07 13:23:52 +00:00
|
|
|
from .vis import *
|
|
|
|
|
2021-08-30 18:50:03 +00:00
|
|
|
__version__ = "0.3.0"
|