tflite_support.task.processor.CategoricalSlot

Represents a categorical slot whose values are within a finite set.

tflite_support.task.processor.CategoricalSlot(
    slot: str,
    prediction: tflite_support.task.processor.Category
)

slot The name of the slot.
prediction The predicted class.

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _CategoricalSlotProto
) -> 'CategoricalSlot'

Creates a CategoricalSlot object from the given protobuf object.

to_pb2

View source

to_pb2() -> _CategoricalSlotProto

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.