julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[REFACTOR] Rename LikelihoodRatioLVQ to SLVQ

Browse Source
This commit is contained in:
Alexander Engelsberger 2021-06-08 08:38:11 +02:00 committed by Alexander Engelsberger
parent e62a8e6582
commit fc11d78b38
No known key found for this signature in database
GPG Key ID: BE3F5909FF0D83E3
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/library.rst
View File

@ -74,7 +74,7 @@ For a test sample they return a distribution instead of a class assignment.
The following two algorihms were presented by :cite:t:`seo2003` .
Every prototypes is a center of a gaussian distribution of its class, generating a mixture model.
.. autoclass:: prototorch.models.probabilistic.LikelihoodRatioLVQ
.. autoclass:: prototorch.models.probabilistic.SLVQ
:members:
.. autoclass:: prototorch.models.probabilistic.RSLVQ

2
prototorch/models/__init__.py
View File

@ -19,7 +19,7 @@ from .glvq import (
)
from .knn import KNN
from .lvq import LVQ1, LVQ21, MedianLVQ
from .probabilistic import CELVQ, RSLVQ, LikelihoodRatioLVQ
from .probabilistic import CELVQ, RSLVQ, SLVQ
from .unsupervised import GrowingNeuralGas, HeskesSOM, KohonenSOM, NeuralGas
from .vis import *

4
prototorch/models/probabilistic.py
View File

@ -61,8 +61,8 @@ class ProbabilisticLVQ(GLVQ):
return loss
class LikelihoodRatioLVQ(ProbabilisticLVQ):
"""Learning Vector Quantization based on Likelihood Ratios."""
class SLVQ(ProbabilisticLVQ):
"""Soft Learning Vector Quantization."""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.loss = LossLayer(nllr_loss)