tflite_support.task.processor.BertCluAnnotationOptions

Options for Bert CLU Annotator processor.

tflite_support.task.processor.BertCluAnnotationOptions(
    max_history_turns: Optional[int] = 5,
    domain_threshold: Optional[float] = 0.5,
    intent_threshold: Optional[float] = 0.5,
    categorical_slot_threshold: Optional[float] = 0.5,
    mentioned_slot_threshold: Optional[float] = 0.5
)

max_history_turns Max number of history turns to encode by the model.
domain_threshold The threshold of domain prediction.
intent_threshold The threshold of intent prediction.
categorical_slot_threshold The threshold of categorical slot prediction.
mentioned_slot_threshold The threshold of mentioned slot prediction.

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _BertCluAnnotationOptionsProto
) -> 'BertCluAnnotationOptions'

Creates a BertCluAnnotationOptions object from the given protobuf object.

to_pb2

View source

to_pb2() -> _BertCluAnnotationOptionsProto

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.

categorical_slot_threshold 0.5
domain_threshold 0.5
intent_threshold 0.5
max_history_turns 5
mentioned_slot_threshold 0.5