tflite_support.task.processor.Pos

Position information of the answer relative to context.

tflite_support.task.processor.Pos(
    start: int, end: int, logit: float
)

It is sortable in descending order based on logit.

start The start offset index of the answer.
end The end offset index of the answer.
logit The logit of the answer. Higher value means it's more likely to be the answer.

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _PosProto
) -> 'Pos'

Creates a Pos object from the given protobuf object.

to_pb2

View source

to_pb2() -> _PosProto

Generates a protobuf object to pass to the C++ layer.

__eq__

View source

__eq__(
    other: Any
) -> bool

Checks if this object is equal to the given object.

Args
other The object to be compared with.

Returns
True if the objects are equal.