ClassifierBase
- class rectools.models.ranking.candidate_ranking.ClassifierBase(*args, **kwargs)[source]
Bases:
ProtocolA protocol that defines the interface for a classifier model. Classes implementing this protocol should provide methods for fitting the model and predicting class probabilities.
- fit()[source]
Fit the classifier to the training data.
- Parameters
args (Any) –
kwargs (Any) –
- Return type
Self
- predict_proba()[source]
Predict class probabilities for the given input data. The implementation should return an array where each element is a probability distribution over the classes.
- Parameters
args (Any) –
kwargs (Any) –
- Return type
ndarray
- Inherited-members
Methods
fit(*args, **kwargs)Fit the classifier to the training data.
predict_proba(*args, **kwargs)Predict class probabilities for the given input data.
- fit(*args: Any, **kwargs: Any) Self[source]
Fit the classifier to the training data.
- Parameters
*args (any) – Positional arguments for fitting the classifier.
**kwargs (any) – Keyword arguments for fitting the classifier.
- Returns
The fitted classifier instance.
- Return type
tpe.Self
- predict_proba(*args: Any, **kwargs: Any) ndarray[source]
Predict class probabilities for the given input data.
- Parameters
*args (any) – Positional arguments for predicting probabilities.
**kwargs (any) – Keyword arguments for predicting probabilities.
- Returns
An array of predicted probabilities, where each element is a distribution over the classes.
- Return type
np.ndarray