tflite_support.task.processor.BertCluAnnotationOptions

Options for Bert CLU Annotator processor.

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

Creates a BertCluAnnotationOptions object from the given protobuf object.

to_pb2

View source

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

__eq__

View source

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