NlpModels Enum

Model names that are supported for NLP (Natural Language Processing) tasks.

Inheritance

Enum

NlpModels

Constructor

NlpModels(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

Fields

BERT_BASE_CASED

BERT_BASE_GERMAN_CASED

BERT_BASE_MULTILINGUAL_CASED

BERT_BASE_UNCASED

BERT_LARGE_CASED

BERT_LARGE_UNCASED

DISTILBERT_BASE_CASED

DISTILBERT_BASE_UNCASED

DISTILROBERTA_BASE

ROBERTA_BASE

ROBERTA_LARGE

XLM_ROBERTA_BASE

XLM_ROBERTA_LARGE

XLNET_BASE_CASED

XLNET_LARGE_CASED