36 lines
907 B
Python
36 lines
907 B
Python
from dataclasses import dataclass
|
|
|
|
from prototorch.models import (
|
|
SimpleComparisonMixin,
|
|
SingleLearningRateMixin,
|
|
SupervisedArchitecture,
|
|
WTACompetitionMixin,
|
|
)
|
|
from prototorch.models.architectures.loss import GLVQLossMixin
|
|
|
|
|
|
class GLVQ(
|
|
SupervisedArchitecture,
|
|
SimpleComparisonMixin,
|
|
GLVQLossMixin,
|
|
WTACompetitionMixin,
|
|
SingleLearningRateMixin,
|
|
):
|
|
"""
|
|
Generalized Learning Vector Quantization (GLVQ)
|
|
|
|
A GLVQ architecture that uses the winner-take-all strategy and the GLVQ loss.
|
|
"""
|
|
|
|
@dataclass
|
|
class HyperParameters(
|
|
SimpleComparisonMixin.HyperParameters,
|
|
SingleLearningRateMixin.HyperParameters,
|
|
GLVQLossMixin.HyperParameters,
|
|
WTACompetitionMixin.HyperParameters,
|
|
SupervisedArchitecture.HyperParameters,
|
|
):
|
|
"""
|
|
No hyperparameters.
|
|
"""
|