Effort selection

EffortSelection

EffortSelection(
    effort_budget: int,
    retrieval_mode: RapidataRetrievalMode = Shuffled,
    max_iterations: int | None = None,
)

Bases: RapidataSelection

With this selection you can define the effort budget you have for a task. As an example, you have a task that takes 10 seconds to complete. The effort budget would be 10.

Parameters:

Name	Type	Description	Default
effort_budget	int	The effort budget for the task.	required
retrieval_mode	RetrievalMode	The retrieval mode for the task.	Shuffled
max_iterations	int | None	The maximum number of iterations for the task.	None

Source code in src/rapidata/rapidata_client/selection/effort_selection.py

def __init__(
    self,
    effort_budget: int,
    retrieval_mode: RapidataRetrievalMode = RapidataRetrievalMode.Shuffled,
    max_iterations: int | None = None,
):
    self.effort_budget = effort_budget
    self.retrieval_mode = retrieval_mode
    self.max_iterations = max_iterations