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` . 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. 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: :members:
.. autoclass:: prototorch.models.probabilistic.RSLVQ .. autoclass:: prototorch.models.probabilistic.RSLVQ

2
prototorch/models/__init__.py
View File

@ -19,7 +19,7 @@ from .glvq import (
) )
from .knn import KNN from .knn import KNN
from .lvq import LVQ1, LVQ21, MedianLVQ 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 .unsupervised import GrowingNeuralGas, HeskesSOM, KohonenSOM, NeuralGas
from .vis import * from .vis import *

4
prototorch/models/probabilistic.py
View File

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