search

keras-team / keras-hub

Pretrained model hub for Keras 3
Apache License 2.0
797 stars 243 forks source link

Add a XLMRobertaClassifier model #414

Closed mattdangerw closed 2 years ago

mattdangerw commented 2 years ago

We should add an XLMRobertaClassifier based on the BertClassifier model. This will wrap the XLMRoberta model with a classification head.

mattdangerw commented 2 years ago

See the following as a template:

https://github.com/keras-team/keras-nlp/blob/master/keras_nlp/models/bert/bert_tasks.py https://github.com/keras-team/keras-nlp/blob/master/keras_nlp/models/bert/bert_tasks_test.py

It might be easiest to wait for https://github.com/keras-team/keras-nlp/issues/405 to start on this.

abheesht17 commented 2 years ago

Working on this.