RecconEmotionEntailmentPostprocessor

class RecconEmotionEntailmentPostprocessor

Class to postprocess RecconEmotionEntailmentModel output to predicted labels

__call__(raw_pred: transformers.modeling_outputs.SequenceClassifierOutput) → List[int]

Convert raw prediction (logits) to predicted label.

Parameters

raw_pred (SequenceClassifierOutput) – output of RecconEmotionEntailmentModel

Returns

list of predicted label

Return type

List[int]