# biome.text.modules.heads.token_classification Module

# TokenClassification Class


class TokenClassification (
    backbone: ModelBackbone,
    labels: List[str],
    label_encoding: Union[str, NoneType] = 'BIOUL',
    top_k: int = 1,
    dropout: Union[float, NoneType] = 0.0,
    feedforward: Union[FeedForwardConfiguration, NoneType] = None,
)

Task head for token classification (NER)

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

# Ancestors

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

# Inherited members

# TokenClassificationConfiguration Class


class TokenClassificationConfiguration (*args, **kwds)

Configuration for classification head components

# Ancestors

# Inherited members

Maintained by