ModelSelectionClassifier¶
A model selector for classification.
Parameters¶
-
models
Type → Iterator[base.Estimator]
-
metric
Type → metrics.base.Metric
Attributes¶
-
best_model
The current best model.
-
models
Methods¶
learn_one
Update the model with a set of features x
and a label y
.
Parameters
- x — 'dict'
- y — 'base.typing.ClfTarget'
Returns
Classifier: self
predict_one
Predict the label of a set of features x
.
Parameters
- x — 'dict'
- kwargs
Returns
base.typing.ClfTarget | None: The predicted label.
predict_proba_one
Predict the probability of each label for a dictionary of features x
.
Parameters
- x
Returns
A dictionary that associates a probability which each label.