Sharat Chikkerur
|
34dd431bc3
|
Adding additional models to text classification
Adding additional models for text classification.
Also adding languages based on the list of currently supported models
['albert-base-v1',
'albert-base-v2',
'albert-large-v1',
'albert-large-v2',
'albert-xlarge-v1',
'albert-xlarge-v2',
'albert-xxlarge-v1',
'albert-xxlarge-v2',
'bart-large',
'bart-large-cnn',
'bart-large-mnli',
'bart-large-xsum',
'bert-base-cased',
'bert-base-cased-finetuned-mrpc',
'bert-base-chinese',
'bert-base-dutch-cased',
'bert-base-finnish-cased-v1',
'bert-base-finnish-uncased-v1',
'bert-base-german-cased',
'bert-base-german-dbmdz-cased',
'bert-base-german-dbmdz-uncased',
'bert-base-japanese',
'bert-base-japanese-char',
'bert-base-japanese-char-whole-word-masking',
'bert-base-japanese-whole-word-masking',
'bert-base-multilingual-cased',
'bert-base-multilingual-uncased',
'bert-base-uncased',
'bert-large-cased',
'bert-large-cased-whole-word-masking',
'bert-large-cased-whole-word-masking-finetuned-squad',
'bert-large-uncased',
'bert-large-uncased-whole-word-masking',
'bert-large-uncased-whole-word-masking-finetuned-squad',
'camembert-base',
'distilbert-base-cased',
'distilbert-base-cased-distilled-squad',
'distilbert-base-german-cased',
'distilbert-base-multilingual-cased',
'distilbert-base-uncased',
'distilbert-base-uncased-distilled-squad',
'distilbert-base-uncased-finetuned-sst-2-english',
'distilroberta-base',
'flaubert-base-cased',
'flaubert-base-uncased',
'flaubert-large-cased',
'flaubert-small-cased',
'mbart-large-en-ro',
'roberta-base',
'roberta-base-openai-detector',
'roberta-large',
'roberta-large-mnli',
'roberta-large-openai-detector',
'umberto-commoncrawl-cased-v1',
'umberto-wikipedia-uncased-v1',
'xlm-clm-ende-1024',
'xlm-clm-enfr-1024',
'xlm-mlm-100-1280',
'xlm-mlm-17-1280',
'xlm-mlm-en-2048',
'xlm-mlm-ende-1024',
'xlm-mlm-enfr-1024',
'xlm-mlm-enro-1024',
'xlm-mlm-tlm-xnli15-1024',
'xlm-mlm-xnli15-1024',
'xlm-roberta-base',
'xlm-roberta-large',
'xlm-roberta-large-finetuned-conll02-dutch',
'xlm-roberta-large-finetuned-conll02-spanish',
'xlm-roberta-large-finetuned-conll03-english',
'xlm-roberta-large-finetuned-conll03-german',
'xlnet-base-cased',
'xlnet-large-cased']
|
2020-06-13 21:21:30 -04:00 |