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[base.typing.FeatureName, Any] - y —
base.typing.ClfTarget
predict_one
Predict the label of a set of features x.
Parameters
- x —
dict[base.typing.FeatureName, Any] - kwargs —
Any
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.