Classification
Path: mozuma.models.classification
This module contains torch modules that can be used as classification heads on top of features.
Models
mozuma.models.classification.modules.LinearClassifierTorchModule
Linear classifier
Attributes:
Name | Type | Description |
---|---|---|
in_features |
int |
Number of dimensions in the input |
label_set |
LabelSet |
The set of labels for this classifier |
mozuma.models.classification.modules.MLPClassifierTorchModule
Multi-layer perceptron classifier
Attributes:
Name | Type | Description |
---|---|---|
in_features |
int |
The number of dimensions in the input |
hidden_layers |
Sequence[int] |
A sequence of width for the hidden layers |
label_set |
LabelSet |
The set of labels for this classifier |
activation |
str | None |
Name of an activation function in the |