tflite_support.task.processor.Mention

A single mention result.

tflite_support.task.processor.Mention(
    value: str, score: float, start: int, end: int
)

value The text value of the mention.
score The score for this mention e.g. (but not necessarily) a probability in [0,1].
start Start of the bytes of this mention.
end Exclusive end of the bytes of this mention.

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _MentionProto
) -> 'Mention'

Creates a Mention object from the given protobuf object.

to_pb2

View source

to_pb2() -> _MentionProto

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.