mp.tasks.decision.DecisionMaker

MediaPipe Decision task.

mp.tasks.decision.DecisionMaker(
    lib: serial_dispatcher.SerialDispatcher, handle: ctypes.c_void_p
)

Methods

close

View source

close()

Releases resources associated with the task.

create_from_model_path

View source

@classmethod
create_from_model_path(
    model_path: str
) -> 'DecisionMaker'

Creates a Decision task from a model file path.

create_from_options

View source

@classmethod
create_from_options(
    options: Optional[DecisionMakerOptions] = None
) -> 'DecisionMaker'

Creates a Decision task from options.

evaluate_batch

View source

evaluate_batch(
    contexts: Sequence[Union[str, DecisionContext]],
    questions: Dict[str, QuestionType]
) -> List[DecisionResult]

Evaluates N independent contexts against M questions in a batch.

evaluate_boolean

View source

evaluate_boolean(
    text: str, question: BooleanQuestion
) -> BooleanResult

Evaluates a Boolean against input text.

evaluate_boolean_batch

View source

evaluate_boolean_batch(
    texts: Sequence[Union[str, DecisionContext]],
    question: BooleanQuestion,
    shared_prefix: Optional[str] = None
) -> List[BooleanResult]

Evaluates a Boolean against a batch of input texts.

evaluate_candidates

View source

evaluate_candidates(
    shared_context: Union[str, DecisionContext],
    candidates: Sequence[Union[str, DecisionContext]],
    questions: Dict[str, QuestionType]
) -> List[DecisionResult]

Evaluates 1 shared prefix + N candidates against M questions.

evaluate_choice

View source

evaluate_choice(
    text: str, question: ChoiceQuestion
) -> ChoiceResult

Evaluates a Choice against input text.

evaluate_choice_batch

View source

evaluate_choice_batch(
    texts: Sequence[Union[str, DecisionContext]],
    question: ChoiceQuestion,
    shared_prefix: Optional[str] = None
) -> List[ChoiceResult]

Evaluates a Choice against a batch of input texts.

evaluate_schema

View source

evaluate_schema(
    text: Union[str, DecisionContext], questions: Mapping[str, QuestionType]
) -> DecisionResult

Evaluates a multi-question schema simultaneously against input text.

evaluate_score

View source

evaluate_score(
    text: str, question: ScoreQuestion
) -> ScoreResult

Evaluates an Score rubric against input text.

evaluate_score_batch

View source

evaluate_score_batch(
    texts: Sequence[Union[str, DecisionContext]],
    question: ScoreQuestion,
    shared_prefix: Optional[str] = None
) -> List[ScoreResult]

Evaluates an Score rubric against a batch of input texts.

prewarm

View source

prewarm(
    question: ChoiceQuestion
) -> None

Prewarms the static prompt prefix and KV cache for a choice question.

prewarm_choice

View source

prewarm_choice(
    question: ChoiceQuestion
) -> None

Deprecated alias for prewarm.

prewarm_schema

View source

prewarm_schema(
    questions: Mapping[str, QuestionType]
) -> None

Prewarms the prompt prefix and caches for a multi-question schema.

__enter__

View source

__enter__()

__exit__

View source

__exit__(
    exc_type, exc_val, exc_tb
)