[WIP] Add CELVQ

TODO Ensure that the distances/probs corresponding to the plabels are sorted
like the target labels.
This commit is contained in:
Jensun Ravichandran
2021-05-27 17:40:16 +02:00
parent 41b2a2f496
commit b7edee02c3
2 changed files with 19 additions and 3 deletions

View File

@@ -2,7 +2,7 @@ from importlib.metadata import PackageNotFoundError, version
from . import probabilistic
from .cbc import CBC, ImageCBC
from .glvq import (GLVQ, GLVQ1, GLVQ21, GMLVQ, GRLVQ, LVQMLN, ImageGLVQ,
from .glvq import (CELVQ, GLVQ, GLVQ1, GLVQ21, GMLVQ, GRLVQ, LVQMLN, ImageGLVQ,
ImageGMLVQ, SiameseGLVQ)
from .lvq import LVQ1, LVQ21, MedianLVQ
from .unsupervised import KNN, NeuralGas