[FEATURE] Add PLVQ model

This commit is contained in:
Alexander Engelsberger
2021-06-08 15:01:08 +02:00
committed by Alexander Engelsberger
parent fc11d78b38
commit c87ed5ba8b
7 changed files with 61 additions and 32 deletions

View File

@@ -2,8 +2,6 @@
Abstract Models
========================================
.. autoclass:: prototorch.models.abstract.AbstractPrototypeModel
.. automodule:: prototorch.models.abstract
:members:
.. autoclass:: prototorch.models.abstract.PrototypeImageModel
:members:
:undoc-members:

View File

@@ -8,7 +8,7 @@ Models
Unsupervised Methods
-----------------------------------------
.. autoclass:: prototorch.models.unsupervised.KNN
.. autoclass:: prototorch.models.knn.KNN
:members:
.. autoclass:: prototorch.models.unsupervised.NeuralGas
@@ -80,9 +80,11 @@ Every prototypes is a center of a gaussian distribution of its class, generating
.. autoclass:: prototorch.models.probabilistic.RSLVQ
:members:
Missing:
:cite:t:`villmann2018` proposed two changes to RSLVQ: First incooperate the winning rank into the prior probability calculation.
And second use divergence as loss function.
- PLVQ
.. autoclass:: prototorch.models.probabilistic.PLVQ
:members:
Classification by Component
--------------------------------------------

View File

@@ -60,3 +60,14 @@
doi = {10.1162/neco.2009.11-08-908},
}
@InProceedings{villmann2018,
author="Villmann, Andrea
and Kaden, Marika
and Saralajew, Sascha
and Villmann, Thomas",
title="Probabilistic Learning Vector Quantization with Cross-Entropy for Probabilistic Class Assignments in Classification Learning",
booktitle="Artificial Intelligence and Soft Computing",
year="2018",
publisher="Springer International Publishing",
}