prototorch_models/prototorch/y_arch/library/glvq.py

36 lines
907 B
Python
Raw Normal View History

2022-05-19 14:13:08 +00:00
from dataclasses import dataclass
2022-05-31 15:56:03 +00:00
from prototorch.y_arch import (
2022-05-19 14:13:08 +00:00
SimpleComparisonMixin,
SingleLearningRateMixin,
SupervisedArchitecture,
WTACompetitionMixin,
)
2022-05-31 15:56:03 +00:00
from prototorch.y_arch.architectures.loss import GLVQLossMixin
2022-05-19 14:13:08 +00:00
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.
"""