# biome.text.modules.heads.classification.text_classification Module

# TextClassification Class


class TextClassification (
    backbone: ModelBackbone,
    labels: List[str],
    pooler: Union[Seq2VecEncoderConfiguration, NoneType] = None,
    feedforward: Union[FeedForwardConfiguration, NoneType] = None,
    multilabel: bool = False,
)

Task head for text classification

Initializes internal Module state, shared by both nn.Module and ScriptModule.

# Ancestors

  • ClassificationHead
  • TaskHead
  • torch.nn.modules.module.Module
  • allennlp.common.registrable.Registrable
  • allennlp.common.from_params.FromParams

# Inherited members

# TextClassificationConfiguration Class


class TextClassificationConfiguration (*args, **kwds)

Configuration for classification head components

# Ancestors

# Inherited members

Maintained by