microsoft / nlp-recipes

Natural Language Processing Best Practices & Examples
MIT License
6.35k stars 915 forks source link

Adding additional models to text classification #597

Closed sharatsc closed 4 years ago

sharatsc commented 4 years ago

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']

Description

Related Issues

Checklist: