Interfaces
Abstract base classes defining standard interfaces for SPFlow modules.
Classifier
Abstract base class for modules that support classification.
-
class spflow.interfaces.classifier.Classifier[source]
Bases: ABC
Abstract base class for modules that support classification.
Provides a standard interface for models that can predict class labels
and class probabilities from input data.
-
predict(data)[source]
Predict class labels for input data.
- Parameters:
data (Tensor) – Input data tensor.
- Return type:
Tensor
- Returns:
Predicted class labels.
-
abstractmethod predict_proba(data)[source]
Predict class probabilities for input data.
- Parameters:
data (Tensor) – Input data tensor.
- Return type:
Tensor
- Returns:
Class probability predictions. Each row corresponds to a data point,
and each column corresponds to a class.