active_learning_steps¶
- probly.evaluation.active_learning.loop.active_learning_steps(pool: ActiveLearningPool, estimator: E, query_strategy: QueryStrategy[E], query_size: int = 1000, n_iterations: int = 10) Iterator[ALState[E]][source]¶
Yield AL state after initial training and each query-retrain cycle.
- Parameters:
pool – Data pool managing labeled/unlabeled splits.
estimator – Model to train and query. Must implement fit/predict/predict_proba.
query_strategy – Strategy for selecting which unlabeled samples to query.
query_size – Number of samples to query per iteration.
n_iterations – Maximum number of query-retrain iterations.
- Yields:
ALState after initial training (iteration=0) and after each subsequent query-retrain cycle.