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[mp.tasks.decision.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: mp.tasks.decision.BooleanQuestion
) -> mp.tasks.decision.BooleanResult

Evaluates a Boolean against input text.

evaluate_boolean_batch

View source

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

Evaluates a Boolean against a batch of input texts.

evaluate_candidates

View source

evaluate_candidates(
    shared_context: Union[str, mp.tasks.decision.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: mp.tasks.decision.ChoiceQuestion
) -> mp.tasks.decision.ChoiceResult

Evaluates a Choice against input text.

evaluate_choice_batch

View source

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

Evaluates a Choice against a batch of input texts.

evaluate_score

View source

evaluate_score(
    text: str,
    question: mp.tasks.decision.ScoreQuestion
) -> mp.tasks.decision.ScoreResult

Evaluates an Score rubric against input text.

evaluate_score_batch

View source

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

Evaluates an Score rubric against a batch of input texts.

prewarm_choice

View source

prewarm_choice(
    question: mp.tasks.decision.ChoiceQuestion
) -> None

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

__enter__

View source

__enter__()

__exit__

View source

__exit__(
    exc_type, exc_val, exc_tb
)